首页 > 解决方案 > python - 如何将csv行的第一项作为python中的列表名称?

问题描述

我有一个这样的 csv 文件,我想采取

A1,R,N,R,U,N,N,R,U,R,R

A2,R,U,R,R,N,N,R,N,N,N

B1,R,N,R,R,R,N,N,N,R,U

现在我想得到一个这样的列表:

A1=[R,N,R,U,N,N,R,U,R,R]

A2=[R,U,R,R,N,N,R,N,N,N]

B1=[R,N,R,R,R,N,N,N,R,U]

我怎样才能做到这一点?

for x in range(len(a)):
        b = a[x].split(",")     
        doc_line.append(b)


for x in range(len(doc_line)):
    print(doc_line[x])

for x in doc_line:
        engine.append(x[0])             
        rel.append(x[1]) 

  

标签: pythondataframecsv

解决方案


您可以从 csv 文件中逐行读取,下面是代码。

from csv import reader

with open('sample.csv', 'r') as csv_file:
    read_csv = reader(csv_file)

    for row in read_csv:
        print(row)

推荐阅读