[英]Selecting portion of multi-dimensional numpy array
我到處搜索,但似乎找不到任何有效的方法來根據索引選擇3d數組的一部分。 舉例來說,假設我有一些尺寸為200 x 200 x 200的3d數組,並且我想選擇和更改所有索引都大於100的所有元素的值
import numpy as np
mask = np.ones((200,200,200))
for x in np.arange(0,mask.shape[0]):
for y in np.arange(0,mask.shape[1]):
for z in np.arange(0,mask.shape[2]):
if x > 100 & y > 100 & z > 100:
mask[x,y,z] = 0
else:
mask[x,y,z] = 1
必須使用np.select或類似方法來執行某些有效的方法,但是我還是無法解決。 任何幫助將非常感激。
我想選擇並更改所有索引都大於100的所有元素的值
請執行以下操作:
mask[101:,101:,101:] = 0
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.