44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
|
import pandas as pd
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
# 设置随机种子
|
||
|
|
np.random.seed(42)
|
||
|
|
|
||
|
|
# 生成企业和设备数据
|
||
|
|
num_rows = 10 # 每个表的行数
|
||
|
|
|
||
|
|
# 构造数据
|
||
|
|
company_ids = np.random.randint(1000, 1100, size=num_rows)
|
||
|
|
device_ids = np.random.randint(100, 200, size=num_rows)
|
||
|
|
material_ids = np.random.randint(0, 100, size=num_rows)
|
||
|
|
product_ids = np.random.randint(0, 200, size=num_rows)
|
||
|
|
|
||
|
|
device_quantities = np.random.randint(50, 200, size=num_rows)
|
||
|
|
material_quantities = np.random.randint(100,200, size=num_rows)
|
||
|
|
product_quantities = np.random.randint(20, 100, size=num_rows)
|
||
|
|
|
||
|
|
# 创建三个表格的数据框
|
||
|
|
df_devices = pd.DataFrame({
|
||
|
|
'企业id': company_ids,
|
||
|
|
'设备id': device_ids,
|
||
|
|
'设备数量': device_quantities
|
||
|
|
})
|
||
|
|
|
||
|
|
df_materials = pd.DataFrame({
|
||
|
|
'企业id': company_ids,
|
||
|
|
'材料id': material_ids,
|
||
|
|
'材料数量': material_quantities
|
||
|
|
})
|
||
|
|
|
||
|
|
df_products = pd.DataFrame({
|
||
|
|
'企业id': company_ids,
|
||
|
|
'产品id': product_ids,
|
||
|
|
'产品数量': product_quantities
|
||
|
|
})
|
||
|
|
|
||
|
|
# 保存为CSV文件
|
||
|
|
df_devices.to_csv('测试数据 companies_devices.csv', index=False)
|
||
|
|
df_materials.to_csv('测试数据 companies_materials.csv', index=False)
|
||
|
|
df_products.to_csv('测试数据 companies_products.csv', index=False)
|
||
|
|
|