52 lines
2.5 KiB
Python
52 lines
2.5 KiB
Python
import pandas as pd
|
||
# 读取Excel文件
|
||
file_path = "/home/xiazj/ai-station-code/meirejie/data/tar_data.csv" # 替换为你的Excel文件路径
|
||
df = pd.read_csv(file_path)
|
||
# 随机抽取10条数据
|
||
test_set = df.sample(n=10, random_state=1) # random_state保证每次抽样结果相同
|
||
|
||
columns = ['A', 'V', 'FC', 'C', 'H', 'N', 'S', 'O', 'H/C', 'O/C', 'N/C', 'Rt','Hr', 'dp', 'T','Tar']
|
||
test_set = test_set[columns]
|
||
# 保存测试集到新的Excel文件
|
||
test_set.to_csv('/home/xiazj/ai-station-code/meirejie/data/tar_data_test.csv', index=False) # 保存为test_set.xlsx，不保存索引
|
||
print("测试集已保存为 char_data_test.csv")
|
||
|
||
|
||
|
||
file_path = "/home/xiazj/ai-station-code/meirejie/data/gas_data.csv" # 替换为你的Excel文件路径
|
||
df = pd.read_csv(file_path)
|
||
# 随机抽取10条数据
|
||
test_set = df.sample(n=10, random_state=1) # random_state保证每次抽样结果相同
|
||
columns = ['A', 'V', 'FC', 'C', 'H', 'N', 'S', 'O', 'H/C', 'O/C', 'N/C', 'Rt','Hr', 'dp', 'T','Gas']
|
||
test_set = test_set[columns]
|
||
# 保存测试集到新的Excel文件
|
||
test_set.to_csv('/home/xiazj/ai-station-code/meirejie/data/gas_data_test.csv', index=False) # 保存为test_set.xlsx，不保存索引
|
||
|
||
print("测试集已保存为 gas_data_test.csv")
|
||
|
||
|
||
|
||
|
||
file_path = "/home/xiazj/ai-station-code/meirejie/data/water_data.csv" # 替换为你的Excel文件路径
|
||
df = pd.read_csv(file_path)
|
||
# 随机抽取10条数据
|
||
test_set = df.sample(n=10, random_state=1) # random_state保证每次抽样结果相同
|
||
columns = ['A', 'V', 'FC', 'C', 'H', 'N', 'S', 'O', 'H/C', 'O/C', 'N/C', 'Rt','Hr', 'dp', 'T','Water']
|
||
test_set = test_set[columns]
|
||
# 保存测试集到新的Excel文件
|
||
test_set.to_csv('/home/xiazj/ai-station-code/meirejie/data/water_data_test.csv', index=False) # 保存为test_set.xlsx，不保存索引
|
||
|
||
print("测试集已保存为 water_data_test.csv")
|
||
|
||
|
||
|
||
file_path = "/home/xiazj/ai-station-code/meirejie/data/char_data.csv" # 替换为你的Excel文件路径
|
||
df = pd.read_csv(file_path)
|
||
# 随机抽取10条数据
|
||
test_set = df.sample(n=10, random_state=1) # random_state保证每次抽样结果相同
|
||
columns = ['A', 'V', 'FC', 'C', 'H', 'N', 'S', 'O', 'H/C', 'O/C', 'N/C', 'Rt','Hr', 'dp', 'T','Char']
|
||
test_set = test_set[columns]
|
||
# 保存测试集到新的Excel文件
|
||
test_set.to_csv('/home/xiazj/ai-station-code/meirejie/data/char_data_test.csv', index=False) # 保存为test_set.xlsx，不保存索引
|
||
|
||
print("测试集已保存为 char_data_test.csv") |