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}")
|