python 逗号分隔_python中如何将数据行拆分成逗号分隔的列

1.numpy解决方案（因为numpy标记）

对numpy数组使用^{}：import numpy as np

arr = np.genfromtxt('file.txt',dtype='str',delimiter=',')

print (arr)

[['red' 'red' 'blue']

['blue' 'red' 'blue']

['blue' 'blue' 'red']]

print (arr[0])

['red' 'red' 'blue']

print (arr[0][2])

blue

2.pandas解决方案：

对DataFrame和选择值^{}使用^{}：import pandas as pd

df = pd.read_csv('file.txt', header=None)

print (df)

0 1 2

0 red red blue

1 blue red blue

2 blue blue red

#select first row to Series

print (df.loc[0])

0 red

1 red

2 blue

Name: 0, dtype: object

#select value by index and column

print (df.loc[0, 2])

blue

3.纯python解决方案：

如果需要嵌套列表，请使用nested list comprehension：data = [[item for item in line.rstrip('\r\n').split(',')]

for line in open('file.txt')]

print (data)

[['red', 'red', 'blue'], ['blue', 'red', 'blue'], ['blue', 'blue', 'red']]

或者使用模块csv：import csv

reader = csv.reader(open("file.txt"), delimiter=',')

data = [word for word in [row for row in reader]]

print (data)

[['red', 'red', 'blue'], ['blue', 'red', 'blue'], ['blue', 'blue', 'red']]print (data[0])

['red', 'red', 'blue']

print (data[0][2])

blue