CSV文件的读写方法

CSV文件的读写方法

1 CSV文件有column

可以使用pandas直接按column读取

import pandas as pd
# 读
df = pd.read_csv(data_path)
df.columns """>>'cat' 'label' 'texts'  """
df['cat'] """>> 返回一个很长的list """
​
# 写：
dataframe = pd.DataFrame({"column1":list1,"column2":list2})
dataframe.to_csv(output_path+name+"file name",index=False,sep=',')
# , 逗号意味着存的是csv文件，如何相隔是\t 则就是tsv文件，修改file name后缀为.tsv

除了按column读，这种也可以按照下面的方法一行一行读

 

2 CSV文件没有column

一行一行读。设置一些list，每一行append以下，最后能读取得到每列数据。

# 读
csvFile = open('../../data/capital/2011-Q3-cabi-trip-history-data.csv','r')
csv_reader = csv.reader(csvFile)
cat = []
label = []
text = []
for one_line in csv_reader:
    # 一行一行的来
    cat.append(one_line[0])
    ...
csv_reader.close()
​
# 写
data = list[]
csvFile2 = open('../../data/capital/0001.csv','w',newline='')
writer = csv.writer(csvFile2)
m = len(data)
for i in range(m):
    writer.writerow(data[i]) # 一行一行写
csvFile2.close()

 

未完带补充...