共計 295 個字符,預(yù)計需要花費(fèi) 1 分鐘才能閱讀完成。
可以使用列表推導(dǎo)式或者 numpy 庫來取出二維數(shù)組的某一列。
- 使用列表推導(dǎo)式:
假設(shè)有一個二維數(shù)組 arr,要取出第 i 列,可以使用以下代碼:
column = [row[i] for row in arr]
其中,i 為要取出的列的索引。
- 使用 numpy 庫:
如果已經(jīng)安裝了 numpy 庫,可以使用 numpy 的切片操作來取出某一列。假設(shè)有一個二維數(shù)組 arr,要取出第 i 列,可以使用以下代碼:
import numpy as np
column = arr[:, i]
其中,i 為要取出的列的索引。注意,numpy 的切片操作中,”:” 表示全部元素。
使用 numpy 庫的方法更加簡潔和高效,尤其對于大型數(shù)組來說。
丸趣 TV 網(wǎng) – 提供最優(yōu)質(zhì)的資源集合!
正文完