shallow iteration with nditer(使用 nditer 进行浅迭代)
本文介绍了使用 nditer 进行浅迭代的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
我有这样一个数组:
>>>y = np.random.randint(0, 255, (2,2,3))
>>>array([[[242, 14, 211],
[198, 7, 0]],
[[235, 60, 81],
[164, 64, 236]]])
而且我必须遍历每个 triplet 元素(不幸的是,矢量化对我没有帮助......).所以我尝试了:
And I have to iterate over each triplet element (unfortunately vectorization won't help me here...). So I tried:
for i, j in np.nditer(y):
print y[i, j],
希望我能得到这样的输出:
hoping I'd get such an output:
[242, 14, 211], [198, 7, 0], [235, 60, 81], [164, 64, 236]
,但运气不好!
我得到错误:
Traceback (most recent call last):
File "<ipython-input-21-a336ef837a8a>", line 1, in <module>
for i, j in np.nditer(y): print y[i,j]
TypeError: iteration over a 0-d array
我很确定我犯了一个非常明显的错误...谁能帮帮我?
I'm quite sure I'm making a very obvious error... can anyone please help me?
推荐答案
看起来你只需要把它拉平.您可以使用 itertools 中的 chain
运算符.
It looks like you just need to flatten this down a level. You can use the chain
operator from itertools.
from itertools import chain
y = np.random.randint(0, 255, (2,2,3)
b = chain.from_iterable(y) # where b is a generator
list(b) 输出
[array([ 51, 119, 84]),
array([ 50, 110, 193]),
array([165, 157, 52]),
array([239, 119, 83])]
这篇关于使用 nditer 进行浅迭代的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!
沃梦达教程
本文标题为:使用 nditer 进行浅迭代
基础教程推荐
猜你喜欢
- 使用PyInstaller后在Windows中打开可执行文件时出错 2022-01-01
- Python kivy 入口点 inflateRest2 无法定位 libpng16-16.dll 2022-01-01
- 用于分类数据的跳跃记号标签 2022-01-01
- 何时使用 os.name、sys.platform 或 platform.system? 2022-01-01
- 如何让 python 脚本监听来自另一个脚本的输入 2022-01-01
- 筛选NumPy数组 2022-01-01
- 如何在海运重新绘制中自定义标题和y标签 2022-01-01
- Dask.array.套用_沿_轴:由于额外的元素([1]),使用dask.array的每一行作为另一个函数的输入失败 2022-01-01
- 线程时出现 msgbox 错误,GUI 块 2022-01-01
- 在 Python 中,如果我在一个“with"中返回.块,文件还会关闭吗? 2022-01-01