python - 将 csv 中的字符串条目写入 python 中的数组？

Question

我正在努力将字符串值放入 python 数组中。我有一个文件，大约有 30k 个条目，每一行看起来像这样:

0R1,Sn=0.3M,Sm=0.7M,Sx=1.5M

我不需要 0R1 部分；我需要的只是一个数组中的所有 Sn 值、另一个数组中的 Sm 值和另一个数组中的 Sx(当然，我还没有想出如何从字符串中获取数值，但是我稍后再考虑)。我想现在我正在尝试制作一个字符串数组。

这是我的代码:

fname = '\\pathname...\\WXT51003.txt'
f1 = open(fname, 'r')

import csv
import numpy
from numpy import zeros
reader = csv.reader(f1)
Max = zeros((29697,1), dtype = numpy.str)
Mean = zeros((29697,1), dtype = numpy.str)
Min = zeros((29697,1), dtype = numpy.str)
for i, row in enumerate(reader):
    Min[i] = row[1]
    Mean[i] = row[2]
    Max[i] = row[3]

f1.close()
print Min[0:10]

print 语句的输出是一个数组，每一行都有 'S'。我如何让它读取整个字符串，而不仅仅是第一个字符？

Answer 1

reader = csv.reader(f1)
rows = list(reader)
cols = zip(*rows)
Min = cols[1]
Mean = cols[2]
Max = cols[3]


# or if you really want numpy.arrays
Min = numpy.array(cols[1]) #dtype will be auto-assigned
Mean = numpy.array(cols[2]) #dtype will be auto-assigned
Max = numpy.array(cols[3]) #dtype will be auto-assigned

我会怎么做...(不要为此使用 numpy ...至少现在还没有)

如果你需要使用 numpy 然后使用一个 dtype 的“S8”或者无论你需要你的字符串有多大......或者使用 dtype.object 或其他......默认情况下它将是一个 len1 字符串类型......但实际上我认为没有理由根据您的代码片段在这里使用 numpy