26 lines
704 B
Python
26 lines
704 B
Python
import pandas as pd
|
||
import numpy as np
|
||
|
||
# 设置数据行数
|
||
total_rows = 100 # 总共100行
|
||
material_count = 61 # 前61行为材料
|
||
|
||
# 生成产品id
|
||
product_ids = np.arange(1, total_rows + 1)
|
||
|
||
# 生成种类，前61行是材料，后面是设备
|
||
categories = ['材料'] * material_count + ['设备'] * (total_rows - material_count)
|
||
|
||
# 创建数据框
|
||
df_products = pd.DataFrame({
|
||
'产品id': product_ids,
|
||
'种类': categories
|
||
})
|
||
|
||
# 保存为CSV文件
|
||
file_path_products = '测试数据 products_materials_equipment.csv'
|
||
df_products.to_csv(file_path_products, index=False) # index=False 不保存行索引
|
||
|
||
# 打印文件路径
|
||
print(f"CSV 文件已生成，路径为: {file_path_products}")
|