多索引Python数组的智能处理

Question

我有以下代码：

p1 = np.array([[[[[[[[[[0.]*2]*2]*2]*2]*2]*2]*2]*2]*2]*2)
s = [0]*10
#
# Do something with s
#
p1[s[0],s[1],s[2],s[3],s[4],s[5],s[6],s[7],s[8],s[9]] += 1

是否有更明智的方法：

1. 创建没有所有方括号的p1，以及
2. 使用字符串或列表访问p1的组件？

我的想法是：

p1[s] += 1

或者：

p1[*s] += 1

例如，如果我想要的不是10个指数，而是N个指数呢？

Answer 1

np.array([[[[[[[[[[0.]*2]*2]*2]*2]*2]*2]*2]*2]*2]*2)

最好写成：

np.zeros((2,2,2,2,2,2,2,2,2,2))

或者因为有十个2：

np.zeros((2,)*10)