修复列表超出范围

Question

我正在尝试从文件中提取一列以将其放入字典中，然后将其转换为数据框，我已尝试过此代码，但它不起作用：

filename ="C:\\Users\Amin Chaari\Desktop\drugs.csv"

with open (filename, encoding="utf8") as f:
datas = csv.reader(f, delimiter = '\t')

lines = [row for row in datas]
R={}
for i in range(0,5):
 print(lines[0][i])
 if lines[0][i] != 'userID':
    if lines[0][i] !='drugName':
        grouped_column = []
        for j,row in enumerate(lines):
            if j>0:
                grouped_column.append(row[i])
                R.update({lines[0][i]:grouped_column})
print(R)
df3=pd.DataFrame(R)

这是我得到的错误：

Traceback (most recent call last):

File "<ipython-input-116-d9d691dce427>", line 9, in <module>
grouped_column.append(row[i])

IndexError: list index out of range

你能帮帮我吗！谢谢

Answer 1

一种“修复”的简单方法是：

R = {}
for i in range(0, 5):
  print(lines[0][i])
  if lines[0][i] != 'userID':
    if lines[0][i] !='drugName':
      grouped_column = []
      for j, row in enumerate(lines):
        try:
          if j > 0:
            grouped_column.append(row[i])
            R.update({lines[0][i]:grouped_column})
        except IndexError:
          pass

print(R)
df3 = pd.DataFrame(R)

这更像是解决您的问题而不是解决方案。您应该调试并查看问题所在。但问题是你的row[i] 不存在，如果你在这里发布整个代码或行中的值会更好。

编辑

此代码将比较并检查该行是否包含需要使用的大小，不区分大小写将忽略该部分：

import csv
import pandas as pd

FILE = "drugs.csv"

def main():

  with open (FILE, encoding="utf8") as f:
    datas = csv.reader(f, delimiter = '\t')
    lines = [row for row in datas]

  R = {}

  for i in range(len(lines[0])):
    if lines[0][i] != 'userID' and lines[0][i] !='drugName':
      grouped_column = []

      for j, row in enumerate(lines):
        if j > 0 and len(row) > i:
          grouped_column.append(row[i])
          R.update({lines[0][i]:grouped_column})

  df3 = pd.DataFrame.from_dict(R, orient="index")
  df3.transpose()


if __name__ == "__main__":
  main()