溫馨提示×

numpy中split函數怎么使用

小億
158
2023-11-15 11:02:29
欄目: 編程語言

在NumPy中,split()函數用于將數組拆分為子數組。split()函數有三個參數:數組、拆分點和軸。

參數解釋:

  • 數組:要拆分的數組。
  • 拆分點:指定在哪些索引位置進行拆分??梢允且粋€整數、一個整數數組或一個下標數組。
  • 軸:指定在哪個軸上進行拆分。默認為0,表示在第一個軸上進行拆分。

下面是使用split()函數的幾個示例:

import numpy as np

# 示例1:將數組拆分成兩個子數組
arr1 = np.array([1, 2, 3, 4, 5, 6])
subarrays = np.split(arr1, 2)
print(subarrays)
# 輸出:[array([1, 2, 3]), array([4, 5, 6])]

# 示例2:將數組拆分成三個子數組
arr2 = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
subarrays = np.split(arr2, 3)
print(subarrays)
# 輸出:[array([[1, 2, 3]]), array([[4, 5, 6]]), array([[7, 8, 9]])]

# 示例3:在指定索引處拆分數組
arr3 = np.array([1, 2, 3, 4, 5, 6])
subarrays = np.split(arr3, [2, 4])
print(subarrays)
# 輸出:[array([1, 2]), array([3, 4]), array([5, 6])]

以上示例展示了split()函數的用法,你可以根據自己的需求靈活使用該函數。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女