我们可以在numpy中做如下切片:
a = np.random.randn(10)
b = a[1: 3] #slice with start=1, end=3
字符串
现在我有了一个开始元组和结束元组,我可以用for
循环得到这些切片:
a = np.random.randn(10)
starts = (1, 2)
ends = (3,4)
all_slices = list()
for s, e in zip(starts, ends):
all_slices.append( a[s:e])
all_slices = np.stack(all_slices, axis=0) # shape (2,2)
型
我可以确保每个切片范围都是相同的:ends[i] - starts[i] = M
。假设len(starts)=len(ends)=N
,我可以在numpy中得到一个没有for
循环的形状为(N,M)的数组吗?
1条答案
按热度按时间snz8szmq1#
linspace
接受数组输入,因此可以生成索引数组字符串
我不得不将参数调整为整数值,并跳过结束点(默认为
linspace
)。型
也许有一种更简单的方法来制作索引数组:
型