读csv文件,返回csv文件的某一列数据。也可获取csv文件的标题
import csv
def read_file_csv(csvfile):
list_data = []
with open(csvfile,encoding='utf-8') as csvfile:
csv_reader = csv.reader(csvfile)
data_header = next(csv_reader)
for row in csv_reader:
list_data.append(row)
list_data = [row[3] for row in list_data]
print(data_header)
return list_data
if __name__ == '__main__':
print(read_file_csv(r"I:\workplace\pythonWorlplace\python310W\precov\precov\func\Wuhan-2019-nCoV.csv")[:10])
列表写入csv
import csv
def write_csv(list_data,csv_file):
with open(csv_file, 'w', newline='', encoding="utf-8") as t:
writer = csv.writer(t)
writer.writerows(list_data)
if __name__ == '__main__':
list_data = [[1,2,3],[1,2,3],[1,2,3],[1,2,3]]
csv_file="test.csv"
write_csv(list_data,csv_file)