簡體   English   中英

numpy,如何在二維數組中找到總行數,在一維數組中找到總列數

[英]numpy, how do I find total rows in a 2D array and total column in a 1D array

您對新手問題表示歉意,但我想知道是否有人可以幫我解決兩個問題。 例子說我有這個,

[[1,2,3],[10,2,2]]

我有兩個問題。

  • 如何查找總列數:
  • 如何查找總行數:

非常感謝你。 一種

獲取行數和列數非常簡單:

>>> import numpy as np
>>> a=np.array([[1,2,3],[10,2,2]])
>>> num_rows, num_cols = a.shape
>>> print num_rows, num_cols
2 3
import numpy as np
a = np.array([[1,2,3],[10,2,2]])
num_rows = np.shape(a)[0]
num_columns = np.shape(a)[1]
>>> import numpy as np
>>> a=np.array([[1,2,3],[10,2,2]])
>>> a
array([[ 1,  2,  3],
       [10,  2,  2]])

#Mean of rows.
>>> np.mean(a,axis=1)
array([ 2.        ,  4.66666667])

#Mean of columns.
>>> np.mean(a,axis=0)
array([ 5.5,  2. ,  2.5])

你也可以用sum來做到這一點:

#Sum of rows.
>>> np.sum(a,axis=1)
array([ 6, 14])

#Sum of columns
>>> np.sum(a,axis=0)
array([11,  4,  5])

Numpy的函數通常采用axis參數,就2D數組而言, axis=0將跨列應用函數,而axis=1將跨行應用此函數。

>>> import numpy as np
>>> a=np.array([[1,2,3],[10,2,2]])
>>> row_count = len(a[:])
>>> col_count = len(a[:][0])
>>> print ("Row_Count:%d   Col_Count:%d " %(row_count,col_count))
Row_Count:2   Col_Count:3

因此,如果你有n維數組,你可以找到所有維度,但你只需要隨后附加[0]

有多種方法可以做到這一點,其中一種方法如下:

import numpy as np
a = np.array([[1,2],[10,20],[30,20]])

# Total Rows: 
np.shape(a)[0]

#Total Columns: 
np.shape(a)[1]

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM