Python:如何在numpy数组中逐行读取?

时间:2016-02-23 13:05:35

标签: python arrays numpy

我想知道我们可以在数组中逐行读取。例如:

array([[ 0.28,  0.22,  0.23,  0.27],
       [ 0.12,  0.29,  0.34,  0.21],
       [ 0.44,  0.56,  0.51,  0.65]])

以数组形式读取第一行以执行某些操作,然后继续第二行数组:

array([0.28,0.22,0.23,0.27])

产生上述数组的原因是由于这两行代码:

from numpy import genfromtxt
single=genfromtxt('single.csv',delimiter=',')

single.csv

0.28,  0.22,  0.23,  0.27
0.12,  0.29,  0.34,  0.21
0.44,  0.56,  0.51,  0.65

使用readlines()看起来像生成列表而不是数组。就我而言,我使用的是csv文件。我试图逐行使用值行而不是一起使用它们以避免内存错误。任何人都可以帮助我吗?

with open('single.csv') as single:
    single=single.readlines()

3 个答案:

答案 0 :(得分:7)

您可以使用np.fromstring

import numpy as np
with open('single.csv') as f:
    lines=f.readlines()
    for line in lines:
        myarray = np.fromstring(line, dtype=float, sep=',')
        print(myarray)

http://docs.scipy.org/doc/numpy-1.10.1/reference/generated/numpy.fromstring.html  和How to read csv into record array in numpy?

答案 1 :(得分:2)

似乎您没有使用Python阅读文件的经验。让我在Ipython迭代会话中详细研究一个例子

创建一个多行文字来模拟你的文件

In [23]: txt="""0.28,  0.22,  0.23,  0.27
0.12,  0.29,  0.34,  0.21
0.44,  0.56,  0.51,  0.65"""

将其拆分为线条以模拟readlines

的结果
In [24]: txt=txt.splitlines(True)

In [25]: txt
Out[25]: 
['0.28,  0.22,  0.23,  0.27\n',
 '0.12,  0.29,  0.34,  0.21\n',
 '0.44,  0.56,  0.51,  0.65']

我可以使用genfromtxt将其转换为数组(您可以将结果传递给readlinesgenfromtxt

In [26]: np.genfromtxt(txt, delimiter=',')
Out[26]: 
array([[ 0.28,  0.22,  0.23,  0.27],
       [ 0.12,  0.29,  0.34,  0.21],
       [ 0.44,  0.56,  0.51,  0.65]])

我可以遍历这些行,剥离\n并拆分''

In [27]: for line in txt:
    print line.strip().split(',')
   ....:     
['0.28', '  0.22', '  0.23', '  0.27']
['0.12', '  0.29', '  0.34', '  0.21']
['0.44', '  0.56', '  0.51', '  0.65']

我可以将每个字符串转换为具有列表解析的浮点数:

In [28]: for line in txt:                                  
    print [float(x) for x in line.strip().split(',')]
   ....:     
[0.28, 0.22, 0.23, 0.27]
[0.12, 0.29, 0.34, 0.21]
[0.44, 0.56, 0.51, 0.65]

或者通过将迭代放在另一个列表解析中,我可以得到一个数字列表列表:

In [29]: data=[[float(x) for x in line.strip().split(',')] for line in  txt]

In [30]: data
Out[30]: [[0.28, 0.22, 0.23, 0.27], [0.12, 0.29, 0.34, 0.21], [0.44, 0.56, 0.51, 0.65]]

我可以把它变成一个数组

In [31]: np.array(data)
Out[31]: 
array([[ 0.28,  0.22,  0.23,  0.27],
       [ 0.12,  0.29,  0.34,  0.21],
       [ 0.44,  0.56,  0.51,  0.65]])

genfromtxt基本上是经过那个序列 - 读取行,分割它们,将字符串转换为值,最后从列表中创建一个数组。

有快捷方式,但我认为您将从详细完成这些步骤中受益。它与基本的Python字符串和列表操作一样多,也与数组有关。

答案 2 :(得分:0)

for list in array:
    print(list)
    for item in list:
        print(item)

给出输出:

[0.28, 0.22, 0.23, 0.27]
0.28
0.22
0.23
0.27
[0.12, 0.29, 0.34, 0.21]
0.12
0.29
0.34
0.21
[0.44, 0.56, 0.51, 0.65]
0.44
0.56
0.51
0.65