numpy,如何在二维数组中找到总行数,在一维数组中找到总列数

时间:2013-09-08 21:45:19

标签: python numpy

您对新手问题表示歉意,但我想知道是否有人可以帮我解决两个问题。 例子说我有这个,

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

我有两个问题。

  • 如何找到总列数:
  • 如何找到总行数:
非常感谢你。 甲

4 个答案:

答案 0 :(得分:8)

获取行数和列数非常简单:

>>> 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

答案 1 :(得分: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]

答案 2 :(得分: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将跨行应用此函数。

答案 3 :(得分:0)

>>> 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]