如何使用python将列中的字符串数据转换为列表

Question

我们需要将列中的字符串数据转换为列表格式。 我们有一个包含7列的csv格式文件，我们需要在类别列中进行更改，并将数据保存到同一个csv文件中

输入数据：

categories
2,3,6
2,4
5
1,5,6
1,2,3

输出：

categories
["2","3","6"]
["2","4"]
["5"]
["1","5","6"]
["1","2","3"]

适用代码：

df = pd.read_csv(r"D:\data.csv")
my_list = np.array(data['categories'].values.tolist())

转换后，我们如何将数据保存到同一个CSV文件中

Answer 1

试试这个：

df = pd.read_csv(r"D:\data.csv", sep=r'\s+')
df['categories'] = df['categories'].str.split(',')
df.to_csv('1.csv', sep='|', index=False)

1.csv文件：

categories
['2', '3', '6']
['2', '4']
['5']
['1', '5', '6']
['1', '2', '3']

Answer 2

# Importing library 
import csv 
  
# data to be written in csv file 
my_list = np.array(data['categories'].values.tolist())
  
# opening the csv file in 'w+' mode 
file = open("D:\data.csv", 'w+', newline ='') 
  
# writing the data into the file 
with file:     
    write = csv.writer(file) 
    write.writerows(my_list)