如何将第一个值放在一列中并保留在另一列中？

Question

ROCO2_CLEF_00001.jpg,C3277934,C0002978 ROCO2_CLEF_00002.jpg、C3265939、C0002942、C2357569

我想从 csv 文件制作一个 pandas 数据框。我想将第一行条目（文件名）放入一列中，并将 column/header 名称设为 "filenames"，并将其余条目放入另一列名称中 "class"。如何操作？

Answer 1

如果您的文件每行没有固定数量的逗号，您可以执行以下操作：

import pandas as pd

csv_path = 'test_csv.csv'
raw_data = open(csv_path).readlines()

# clean rows
raw_data = [x.strip().replace("'", "") for x in raw_data]
print(raw_data)

# make split between data
raw_data = [ [x.split(",")[0], ','.join(x.split(",")[1:])] for x in raw_data]
print(raw_data)

# build the pandas Dataframe
column_names = ["filenames", "class"]
temp_df = pd.DataFrame(data=raw_data, columns=column_names)

print(temp_df)

              filenames                       class
0  ROCO2_CLEF_00001.jpg           C3277934,C0002978
1  ROCO2_CLEF_00002.jpg  C3265939,C0002942,C2357569

如何将第一个值放在一列中并保留在另一列中？

how to put first value in one column and remaining into other column?

pandas

sklearn-pandas