matplotlib读取csv文件
时间: 2023-06-05 19:47:39 浏览: 413
Python中处理数据可视化时,常用的一种库便是matplotlib。matplotlib支持从多种数据源中读取数据,其中包括CSV文件。下面介绍如何使用matplotlib读取CSV文件。
在Python中使用matplotlib读取CSV文件的步骤如下:
1.导入必要的模块
首先需要导入numpy模块和matplotlib模块,以及引入CSV库。
```
import numpy as np
import matplotlib.pyplot as plt
import csv
```
2.读取CSV文件
用CSV库中的reader函数来读取CSV文件。假设文件如下:
```
Year,Sales
2010,10000
2011,15000
2012,18000
2013,20000
2014,25000
2015,30000
```
读取并打印CSV文件中的内容代码如下:
```
with open('sales.csv', 'r') as csvfile:
data = csv.reader(csvfile)
for row in data:
print(row)
```
3.将读取的CSV数据转换为数组
CSV数据读取出来是文本形式的,需要将其转换为数组。下面的代码将读取的数据转换为numpy数组。
```
with open('sales.csv', 'r') as csvfile:
data = csv.reader(csvfile)
# 跳过表头
next(data)
x = []
y = []
for row in data:
x.append(int(row[0]))
y.append(int(row[1]))
x = np.array(x)
y = np.array(y)
```
4.绘制图形
现在已经将CSV数据成功地转换成数组,可以使用matplotlib绘制图形了。以下代码绘制了一个折线图:
```
plt.plot(x, y, marker='o')
plt.title('Sales')
plt.xlabel('Year')
plt.ylabel('Sales')
plt.show()
```
上述代码将年份表示为X轴,销售额表示为Y轴,并标出标题、X轴标签和Y轴标签。最后使用plt.show()显示绘制出的图形。
总的来说,使用matplotlib读取CSV文件的过程就是将CSV文件转换为数组,并使用matplotlib绘制图形。
阅读全文