Python索引错误（超出范围）访问二维数组中的元素

Question

我正在尝试访问从 csv 文件中的数据创建的二维数组的元素。我可以很好地打印数组。

当我尝试访问数组以查找某个元素（即来自 'row' 1 'column' 5 的数字）时，它会引发错误：

C:\Users\AClayton\Current\python begin\code_tester.py in create_alldata(whichfile)
37     array_data=np.array(all_data)
---> 38     nb=array_data[1][5]
IndexError: index 1 is out of bounds for axis 0 with size 1

如果有人能帮忙就太好了

def create_alldata(whichfile):
    open_file = open(infile, 'rb')                       
    csv_current=csv.reader(open_file)                     
    all_data=[]                              
    np.array(all_data)
    for row in open_file:         
        all_data.append(row)                            
    open_file.close()
    array_data=np.array(all_data)
    nb=array_data[1][5]
    return array_data,    


path=raw_input('What is the directory?')
for infile in glob.glob(os.path.join(path, '*.csv')): 
    create_alldata(infile)

Answer 1

如果您想从 CSV 读取多维数据，请使用 numpy.genfromtxt() 或 numpy.loadtxt() 函数，具体取决于您的 CSV 文件的完整性（如果行长变化则使用前者，如果行长不变则使用后者）。

您正在尝试手动构建一个多维 numpy 数组，但您发现它并不像那样工作。

import numpy

def create_alldata(whichfile):
    return numpy.genfromtxt(whichfile)