如何使用逗号读取csv字段并将字段中的字符串拆分为列表？

Question

I am trying to read csv and split some output string with delimiter into a list, but when I tried to read the list using array, it threw error saying "list index out of range". The list should have 2 elements. Please, see attached codes. Please, kindly help me pointing out what was wrong with the code. The index 0 works just fine.

The csv file look like this: ID, Full Name, Last First Name

1, John Smith, "Smith, Jonh"

2, Camil Johnson, "Johnson, Camil"

3, Bang Dong, "Dong, Bang"

4, Charles Simpson, "Simpson, Charles"

import csv
with open('testfile.csv', 'r', newline='') as f:
    reader = csv.reader(f, delimiter=',')
    for row in reader:
        # print(row)
        # print (row[2])
        a = row[2].split(sep=", ")
        print(a[1])

Answer 1

You might try this:

import pandas as pd

df = pd.read_csv('testfile.csv')

This will read in your csv file into a data frame, and you can access the columns and convert them to lists like this:

df['Full Name'].tolist()

Answer 2

The problem is your are using "," as a delimiter, but your "Last First Name" Column also contains commas.

In this line of code:

a = row[2].split(sep=", ")

you are trying to separate the first name from the last name and assume it returns a tuple which you are trying to display here: print(a[1])

The Problem is a will not be a tuple, because of the comma thats inside of this column. Instead there will be a row[3], in your case "Smith" will be in row[2] and "John" in row[3]. To avoid this, use a different delimiter .

Answer 3

Using pandas the same way I coded with csv works very well. Thanks Nathaniel.

import pandas as pd
df = pd.read_csv('testfile.csv')
a = df['Last First Name'].tolist()
for row in a:
    b = row.split(', ')
    print(b[1])