在处理数据的过程中时常会遇到csv(逗号分隔值)文件,其以纯文本形式存储表格数据(数字和文本),现在列出本人用python编写的读写csv文件的示例代码。
1、读取csv文件列表数据
需要读取的表格信息如下:
python代码如下:
# -*- coding:UTF-8 -*-
import pandas as pd
csv_path = 'test.csv'
datas = pd.read_csv(csv_path)
for index in range(len(datas)):
num = datas.loc[index][0]
name = datas.loc[index][1]
grade = datas.loc[index][2]
print("num:", num, "name:", name, "grade:" ,grade)
注:若遇到中文字符,pd.read_csv(file_name, encoding='utf-8')
2、写入数据到csv文件
现在参考1中的csv文件数据,写入相同的数据到csv文件中,示例代码如下:
csv_path1 = 'test1.csv'
dict_lists = []
dict_data1 = {'number':1, 'name': u'Tom', 'grade': 80}
dict_data2 = {'number': 2, 'name': u'Alice', 'grade': 90}
dict_data3 = {'number': 3, 'name': u'Jone', 'grade': 96}
dict_data4 = {'number': 4, 'name': u'Mark', 'grade': 76}
dict_lists.append(dict_data1)
dict_lists.append(dict_data2)
dict_lists.append(dict_data3)
dict_lists.append(dict_data4)
df = pd.DataFrame(dict_lists, columns=['number', 'name', 'grade'])
df.to_csv(csv_path1, index = False)
注:若遇到中文字符,pd.to_csv(file_name, index = False,encoding='utf-8')
“index = True”表示加入索引,默认index = True的