如何从 csv 文件中提取特定列并使用 python 绘图答案

【问题标题】：how to extract speciific columns from a csv file and plot using python如何从 csv 文件中提取特定列并使用 python 绘图
【发布时间】：2012-07-21 14:43:59
【问题描述】：

我有一个 csv 文件，其中包含以下几行数据：

# Vertex        X                Y                Z               K_I             K_II             K_III              J
  0       2.100000e+00      2.000000e+00     -1.000000e-04    0.000000e+00   0.000000e+00        0.000000e+00    0.000000e+00    
  1       2.100000e+00      2.000000e+00      1.699733e-01    8.727065e+00  -8.696262e-04        -1.800691e-04   3.465355e-04    
  2       2.100000e+00      2.000000e+00      3.367067e-01    8.907810e+00  -2.548819e-04        -2.789738e-04   3.610383e-04

我想绘图：

顶点（第一列）与 K_I（第五列）

顶点（第一列）与 K_II（第六列）

顶点（第一列）与 K_III（第七列）

如何从该文件中提取特定列，然后从这些列中绘制折线图？非常感谢任何帮助！

【问题讨论】：

标签： python csv plot

【解决方案1】：

为了让您开始，您可以查找用于处理 CSVs 的 CSV 库（以及制表符和其他分隔文件。有几个选项可以绘制数据，但 matplotlib 是您可能想看看的绝佳选择。

虽然不完全是您想要的，但我在 plotting from SQL Server using matplotlib and PyQT 上的一篇文章刚刚发布，其中包括一些使用 PyQT Gui 在 matplotlib 中绘图的相关示例。不过，我没有解决从 CSV 中提取数据的问题。

【讨论】：

【解决方案2】：

pandas 具有内置的可视化支持，这使得这项任务非常简单。它比 csv + matplotlib 更直接，因为 pandas 为您抽象了它。

【讨论】：

【解决方案3】：

import pandas as pd
frame = pd.read_csv('filename.csv', index_col=0)
frame.K_I.plot() # Vertex is the x-axis

“frame.plot()”也绘制所有列

【讨论】：

我收到一条错误消息：AttributeError: 'DataFrame' object has no attribute 'K_I'