蟒蛇中的二维数组/列表

Question

我试图将一个csv读入二维数组中，而且我似乎被卡住了。任何帮助下面的代码将不胜感激。

import sys, csv 

rowColumn = [] 
rowColumn.append([]) 
rowColumn.append([]) 

with open('filelocation.csv', "r+") as inputFile: 
    fieldnames = ['col1','col2','col3','col4','col5','col6','col7','col8',] 
    reader = csv.reader(inputFile) 
     i1=-1 
     i2=0 
     for row in reader: 
      i1=i1+1 
      print(row) 
      for item in row: 
       #store item in 2D list/array 
       rowColumn[i1][i2].append(item)#error occurs here 
       i2=i1+1 

工作代码：

import sys, csv 

rowColumn = [] 
rowColumn.append([]) 
rowColumn.append([]) 

with open('filelocation.csv', "r+") as inputFile: 
    reader = csv.reader(inputFile) 
    i1=-1 
    for row in reader: 
     i1+=1 
     i2=0 
     for item in row: 
      #store item in 2D list/array 
      rowColumn[i1][i2].append(item) 
      i2+=1 

看来我i2变量被设在了错误的地方。

Answer 1

你的问题是在最后一行：你让你的列号取决于行号。尝试这样的事情，而不是：

for row in reader: 
     i1 += 1 
     print(row) 
     i2 = 0 
     for item in row: 
      #store item in 2D list/array 
      rowColumn[i1][i2].append(item) 
      i2 += 1 

在你以前没有见过的情况下，

我+ = 1

等同于

I = i + 1

。

Answer 2

它是很难被击败大熊猫的易用性：

import pandas as pd 

# Read CSV file into pandas dataframe df 
df = pd.read_csv('filelocation.csv', index_col=False) 

# Convert dataframe to python list via a numpy array 
python_list = df.values.tolist()