尝试访问python脚本中的列时获取行

我有一个问题，似乎答案很容易解释。 我正在使用以下代码从包含行和列的csv文件中升级：

import csv
import os
import glob
import numpy as np

def get_data(filename):
    with open(filename, 'r') as f:
        reader = csv.reader(f)                      
        return list(reader)

all_data = []

path=raw_input('What is the directory?')       
for infile in glob.glob(os.path.join(path, '*.csv')):
     all_data.extend(get_data(infile))
a = np.array(all_data)
col=a[:,[0,1]]
print col

---

但是，结果生成的是前两行而不是列？

问题可能是您的数据。验证是否正确读取CSV数据（调试读卡器）

当我使用以下输入文件运行脚本时：

file1.csv

foo,bar,baz
1,2,3
2,3,4
3,4,5

文件2.csv

4,5,6
5,6,7
6,7,8

然后，脚本的结果如下所示：

$ python test.py
[['4' '5']
 ['5' '6']
 ['6' '7']
 ['foo' 'bar']
 ['1' '2']
 ['2' '3']
 ['3' '4']]

---

因此，我认为您的错误一定是关于读取CSV文件。可能列分隔符工作不正常。

非常感谢，现在一次运行一个文件，似乎可以正常工作