擅长:python、mysql、java
<p>您可能需要使用<code>numpy</code>来完成此操作。您可以使用<a href="https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.r_.html" rel="nofollow noreferrer">^{<cd2>}</a>连接多个切片,并使用它们对数组进行切片分配:</p>
<pre><code>import numpy as np
n = 10
a = np.zeros(n)
a[np.r_[0,n//2]] = 1
a[np.r_[1:n//2]] = 2
a[np.r_[n//2+1:n]] = 0
print(a)
array([1., 2., 2., 2., 2., 1., 0., 0., 0., 0.])
</code></pre>