获取字符串列表并根据逗号将其拆分为二维列表

Question

到目前为止我有以下代码：

def display2dlist(data):
    for i in data:               #loops through list
        lines = i.strip()       #removing leading and trailing characters
        split_lines = lines.split(", ")       #splitting lines by ','
        print(split_lines)       #prints list


with open("python.csv", "r") as infile:
    data = infile.readlines()[0:]       #reads lines from file 
#print(data)
display2dlist(data)       #calling the function display2dlist with data list as parameter

它从 csv 文件中读取行并将每一行保存为列表。它输出这个：

['2016,London,10']
['2017,Tokyo,11']
['2018,Toronto,12']
['2018,Dubai,23']

我该怎么做而不是将每一行都保存为一个大字符串。它在每个逗号处拆分行并保存多个值。所以它应该是这样的：

['2016','London','10']
['2017','Tokyo','11']
['2018','Toronto','12']
['2018','Dubai','23']

例如在我当前的代码中：

data[0][0]= '2'
data[0][1]= '0'        #it is treating it as a big string and just going letter by letter
data[0][2]= '1'
data[1][0]= '2'
data[1][1]= '0'         
data[1][2]= '1'

我想要它，所以当我执行前面的代码时，输出是：

data[0][0]='2016'
date[0][1]='London'
data[0][2]='10'
data[1][0]='2017'
data[1][1]='Tokyo'         
data[1][2]='11'

Answer 1

在 split 方法中删除多余的 space 并使用此函数获取嵌套列表：

def get_2d(data):
    listdata = []
    for i in data:
        split_lines = i.strip().split(",") #  not ", "
        listdata.append(split_lines)
    return listdata

...

with open("python.csv", "r") as infile:
    data = infile.readlines()[0:] 
print(get_2d (data))

Answer 2

def split_data(data):  #data given as list 
    return [i.split(',') for i in data]  #if data is not in list, wrap this in []


#example:
my_data = ['2020,10,10']

split_data(my_data)

#returns
[['2020', '10', '10']]

获取字符串列表并根据逗号将其拆分为二维列表

Take list of strings and split it based on commas into 2d list

python

split

list

readfile