文章/答案/技术大牛

发布

社区首页 >问答首页 >在python中迭代Numpy数组行，即使数组为1行/2列

问在python中迭代Numpy数组行，即使数组为1行/2列
EN

Stack Overflow用户

提问于 2020-06-02 05:00:19

回答 1查看 354关注 0票数 3

我正在尝试迭代一个numpy数组的行。该数组可能像[[a, b], [c, d], ...]一样由两列和多行组成，或者有时像[a, b]那样由单行组成。

对于一维数组，当我使用enumerate遍历各行时，python会生成单个元素a，然后是b，而不是一次性生成完整的行[a, b]。

如何以与2D情况相同的方式迭代一维情况？

row

enumerate

python

arrays

numpy

回答 1

Stack Overflow用户

回答已采纳

发布于 2020-06-02 05:24:44

无论如何，Numpy都会在第一维上迭代。在迭代之前检查形状。

>>> x = np.array([1, 2])
>>> x.ndim
1

>>> y = np.array([[1, 2], [3, 4], [5, 6]])
>>> y.ndim
2

最简单的方法可能是始终包装在对np.array的调用中

>>> x = np.array(x, ndmin=2, copy=False)
>>> y = np.array(y, ndmin=2, copy=False)

这将根据需要将形状1的维度预先添加到数组中。它的优点是，您的输入甚至不必是数组，只需将其转换为数组即可。

另一种选择是使用atleast_2d函数：

>>> x = np.atleast_2d(x)

尽管如此，您很可能会因为尝试一个普通的python循环而牺牲了使用numpy的大部分好处。相反，请尝试向量化您的操作。

票数 2

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/62140913

复制

相似问题

问在python中迭代Numpy数组行，即使数组为1行/2列
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问在python中迭代Numpy数组行，即使数组为1行/2列EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问在python中迭代Numpy数组行，即使数组为1行/2列
EN