溫馨提示×

python怎么取出二維數(shù)組的某一列

小億
547
2023-09-22 19:26:19
欄目: 編程語言

可以使用列表推導(dǎo)式或者numpy庫來取出二維數(shù)組的某一列。

  1. 使用列表推導(dǎo)式:

假設(shè)有一個(gè)二維數(shù)組arr,要取出第i列,可以使用以下代碼:

column = [row[i] for row in arr]

其中,i為要取出的列的索引。

  1. 使用numpy庫:

如果已經(jīng)安裝了numpy庫,可以使用numpy的切片操作來取出某一列。假設(shè)有一個(gè)二維數(shù)組arr,要取出第i列,可以使用以下代碼:

import numpy as np
column = arr[:, i]

其中,i為要取出的列的索引。注意,numpy的切片操作中,":"表示全部元素。

使用numpy庫的方法更加簡潔和高效,尤其對于大型數(shù)組來說。

0