python读取csv文件,存入一维数组（列表转数组），并绘制图像

夜曲2021_

已于 2022-03-18 13:36:27 修改

阅读量6.8k

点赞数 6

文章标签： python 开发语言

于 2022-03-18 13:34:52 首次发布

本文链接：https://blog.csdn.net/fightclub_/article/details/123567853

版权

该博客介绍了如何使用Python的csv和matplotlib库从CSV文件中读取数据并绘制图像。首先，按行读取文件内容，将数据转换为浮点数数组并绘制一维图像。接着，按列读取数据，将指定列转换为数字列表，再进行数据分割并绘制红色一维数组图像。这两部分都展示了将CSV数据转化为可视化图像的基本步骤。

摘要由CSDN通过智能技术生成

按行读取，并绘制图像：

import csv
import matplotlib.pyplot as plt
import numpy as np

with open('filename.csv') as f:                #读取文件
    reader = csv.reader(f)                     #创建阅读器
    rows = [row for row in reader]             #按行读取
    print(rows[0])                             #打印第0行，此时读取的数据类型为列表
y =np.array(rows[0],dtype=np.float32)          #将列表转为数组，数据类型为浮点数
print(y)                                       #打印数组，验证数据类型转换成功
#绘制一维数组图像
t = np.linspace(0,17,17)                       #创建x轴
plt.plot(t,y)                                  #绘制图像
plt.show()

按列读取，并绘制图像：

with open('filename.csv') as f:                
    reader = csv.reader(f)
    weizhen = []                            #创建存储列数据的空列表
    for row in reader:                      #遍历文件中的每一行
        data = float(row[1])                #将索引1处即第二列的字符串数据转换为数字
        weizhen.append(data)                #将数据存储到列表中
    print(weizhen)                          #验证一下，此时类型应为数组
a = weizhen[1001: 1201]                     #数据分割
#绘制一维数组
fig = plt.figure(dpi=128, figsize=(10, 6))
plt.plot(a, c='red')
plt.show()