将文件元素读入3个不同的数组

时间:2014-03-12 13:44:00

标签: python arrays csv file-io

我有一个用x,y,x的空格分隔的文件。我需要可视化数据,所以我想我需要将文件读入3个独立的数组(X,Y,Z),然后绘制它们。我如何将文件读入3个单独的数组,到目前为止我已经删除了每一行末尾的空白元素。

def fread(f=None):
    """Reads in test and training CSVs."""
    X = []
    Y = []
    Z = []

    if (f==None):
        print("No file given to read, exiting...")
        sys.exit(1)

    read = csv.reader(open(f,'r'),delimiter = ' ')
    for line in read:
        line = line[:-1]

我尝试添加类似的内容:

for x,y,z in line:
    X.append(x)
    Y.append(y)
    Z.append(z)

但是我收到类似“ValueError:解压缩的值太多”的错误

我做了很多谷歌搜索,但似乎没有什么可以解决必须在一个文件中读入每个元素的单独数组。

我应该添加我的数据没有很好地排序到行/列中它看起来像这样 “107745590026 2 0.02934046648 0.01023879368 3.331810236 2 0.02727724425 0.07867902517 3.319272757 2 0.01784882881”......

谢谢!

1 个答案:

答案 0 :(得分:1)

编辑:如果您的数据实际上没有分成3个元素的行(而是一个长的空格分隔的值列表),您可以使用python list slicing with stride来实现这一点更容易:

X = read[::3]
Y = read[1::3]
Z = read[2::3]

可能会发生此错误,因为read中的某些行包含三个以上以空格分隔的值。从你的问题中不清楚你在这些情况下想要做什么。如果你正在使用python 3,你可以将一行的第一个元素放入X,将第二个元素放入Y,将该行的所有其余部分放入Z,其中包含以下内容:

for x, y, *z in line:
    X.append(x)
    Y.append(y)
    for elem in z:
        Z.append(elem)

如果您没有使用python 3,则可以稍微冗长的方式执行相同的基本逻辑:

for i, elem in line:
    if i == 0:
        X.append(elem)
    elif i == 1:
        Y.append(elem)
    else:
        Z.append(elem)