NumPy涓€庝箞鍒嗗壊鏁扮粍
鍦∟umPy涓紝鍙互浣跨敤numpy.split()
鍑芥暟鏉ュ垎鍓叉暟缁勩€傝鍑芥暟鎺ュ彈涓変釜鍙傛暟锛屽垎鍒负寰呭垎鍓茬殑鏁扮粍銆佸垎鍓茬偣鎴栧垎鍓茬殑浣嶇疆銆佹部鐫€鍝釜杞磋繘琛屽垎鍓层€?/p>
绀轰緥濡備笅锛?/p>
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
# 灏嗘暟缁刟rr鍒嗗壊鎴愪袱閮ㄥ垎
splitted_arr = np.split(arr, [3])
print(splitted_arr)
杈撳嚭缁撴灉涓猴細
[array([1, 2, 3]), array([ 4, 5, 6, 7, 8, 9, 10])]
鍦ㄤ笂闈㈢殑绀轰緥涓紝鎴戜滑灏嗘暟缁?code>arr鎸夌収绱㈠紩3杩涜鍒嗗壊锛屽緱鍒颁簡涓や釜瀛愭暟缁勩€?/p>