如何保存某些列从csv文件作为Python中的变量？[重复]

Question

我想使用matplotlib在Python中保存和绘制某些列。列的参数将从cmdline获得，因此我必须使用sys。argv来获取它们。以下是我目前拥有的：

编辑：我还应该提到，列号可以根据用户选择的内容而变化。例如，它们可以只执行列1、2或列1。

with open('./P14_data.csv', 'rb') as csvfile:
    data = csv.reader(csvfile, delimiter=';')
    cols = [index for index in sys.argv[1:]]

    #I want to extract the columns corresponding to cols

    for col in cols:

    x[col] = [rows for rows in data]
    print x

但这将返回一个空列表[]。

至于输出，我想将每一列绘制为一维数组。例如，使用表单的csv文件：

1 5 
1 3
0 2
0 3
1 1
1 3

如果用户输入1，我希望我的代码在数组中只保存一列变量：

data=[[1,1,0,0，…]

plt.plot（数据）

我知道熊猫是一个有效的选择，但我喜欢先这样学习。谢谢！




Answer 1

你可以试试这样的东西：

#!/usr/bin/env python3

import csv

with open('./P14_data.csv', newline='') as csvfile:
    data = csv.reader(csvfile, delimiter=';')
    x = [rows for rows in data]
    transposed = list(zip(*x))
    print(transposed)

或者更简单：

#!/usr/bin/env python3

import csv

with open('./P14_data.csv', newline='') as csvfile:
    data = csv.reader(csvfile, delimiter=';')
    transposed = list(zip(*data))
    print(transposed)

要点：

• zip（）

Answer 2

下面是一种将标题和数据保存在单独数组中的方法。为了得到每个列，我们采用数据数组的转置，并选择我们感兴趣的列。

以下是data.csv：

index,value1,value2
0,10,20
1,12,18
2,5,6
3,9,10
4,11,8

代码如下：

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

with open('data.csv','r') as csvfile:
    r = csv.reader(csvfile, delimiter=',')
    data = [i for i in r]

headings = data.pop(0)
data = np.array([[np.float(j) for j in i] for i in data])

c1 = data.T[1]
c2 = data.T[2]

fig, ax = plt.subplots(1)
ax.plot(c1, label=headings[1])
ax.plot(c2, label=headings[2])
ax.legend()
fig.show()

Answer 3

你可以得到行中的第一列

with open('./P14_data.csv', 'rb') as csvfile:
    data = csv.reader(csvfile, delimiter=';')
    included_cols = [1,2,3]
    x = [[rows[0]] for rows in data]
    x1 = [[row[0].split(',')[0] for row in x]]
    x2 = [[row[0].split(',')[1] for row in x]]
    x3 = [[row[0].split(',')[2] for row in x]]

    print x1
    # [['4', '7', '3', '3']]

    print x2
    # [['9', '11', '5', '6']]

    print x3
    # [['5', '4', '2', '3']]