首先需要了解Python的csv模块,该模块为读取Comma-Separated Values(逗号分隔值)类型的文件提供了支持。
1. 导入csv模块
import csv
2. 读取CSV文件
首先需要获取CSV文件的路径,可以使用Python的os模块获取当前目录下的CSV文件的路径。
import os
csv_file_path = os.path.join(os.getcwd(), 'example.csv')
接着,使用csv模块的reader函数读取CSV文件的内容。
with open(csv_file_path, 'r') as file:
reader = csv.reader(file)
for row in reader:
print(row)
以上代码中, 'r' 表示只读模式打开文件,并创建一个csv.reader对象。 csv.reader对象允许对CSV文件的行进行迭代,并返回一个由该行所有值组成的列表。
3. 示例1:打印CSV文件内容
假设CSV文件的内容如下所示:
name, age, gender
Tom, 24, M
Jerry, 29, M
Lily, 22, F
以下代码读取以上文件并打印其内容:
import csv
import os
csv_file_path = os.path.join(os.getcwd(), 'example.csv')
with open(csv_file_path, 'r') as file:
reader = csv.reader(file)
for row in reader:
print(row)
输出结果:
['name', ' age', ' gender']
['Tom', ' 24', ' M']
['Jerry', ' 29', ' M']
['Lily', ' 22', ' F']
4. 示例2:绘制CSV文件中某一列数据的图表
假设CSV文件的内容如下所示:
name, age, gender
Tom, 24, M
Jerry, 29, M
Lily, 22, F
以下代码读取以上文件并绘制年龄的柱状图:
import csv
import os
import matplotlib.pyplot as plt
csv_file_path = os.path.join(os.getcwd(), 'example.csv')
ages = []
with open(csv_file_path, 'r') as file:
reader = csv.DictReader(file)
for row in reader:
ages.append(int(row['age']))
plt.bar(range(len(ages)), ages)
plt.show()
输出结果为一个柱状图,表示CSV文件中的人员年龄。
综上所述,要读取当前目录下的CSV文件,需要导入csv模块,使用os模块获取CSV文件路径并使用csv.reader函数读取CSV文件内容。若需要对读取到的文本数据进行可视化处理,则需要使用相关库,如matplotlib等。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:python读取当前目录下的CSV文件数据 - Python技术站