首页 > 解决方案 > Numpy数组以及如何赋值

问题描述

使用 3D 数组。在嵌套 for 循环中,使用 if 条件将 3D 数组中的所有值分配为等于一个给定值。

import numpy as np 
Topvalue = 1000
center = 5
dim = 2
Bgr = 4
err = 0
#
x = np.ones(dim)
y = np.ones(dim)
z = np.ones(dim)
A =np.array(np.ones((dim,dim,dim)))*Bgr

Rin=[9.89/2]
Rout=[11.89/2]

for i in range (len(x)):
    for j in range(len(y)):
        for k in range(len(z)):
            print(A[i][j][k]=0)
            value = (np.sqrt(np.power(((i-center)/2),2) + np.power(((j-center)/2),2) + np.power(((k-center)/2),2)))
            if value >(Rin[0]-err):
                A[i][j][k] == Topvalue #np.where(A = np.ones((dim,dim,dim))*Bgr, Topvalue, A)    
                if value >=(Rin[0]-err) and (value<Rout[0]):
                    A[i][j][k]==0
                    print(A[i][j][k])

SyntaxError:关键字不能是表达式

标签: pythonarraysnumpy

解决方案


推荐阅读