Бих искал да направя нещо подобно:
if dim==2:
a,b=grid_shape
for i in range(a):
for j in range(b):
A[i,j] = ...things...
където dim
е просто броят на елементите в моя кортеж grid_shape
. A
е numpy масив с измерение dim
. Има ли начин да го направите, без да сте специфични за измерението? Без да се налага да пишете грозен код като
if dim==2:
a,b=grid_shape
for i in range(a):
for j in range(b):
A[i,j] = ...things...
if dim==3:
a,b,c=grid_shape
for i in range(a):
for j in range(b):
for k in range(c):
A[i,j,k] = ...things...