Python：如何在 numpy 数组中逐行读取？

Question

我想知道我们可以在数组中逐行读取吗？例如：

 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=',')

单.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()

原文由 Xiong89 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

你可以使用 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?

原文由 maazza 发布，翻译遵循 CC BY-SA 3.0 许可协议

Answer 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 将其转换为数组（您可以像这样将结果传递给 readlines 到 genfromtxt 。

 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 字符串和列表操作的练习，也是关于数组的练习。

原文由 hpaulj 发布，翻译遵循 CC BY-SA 3.0 许可协议