如何在循环中使用python矩阵的维度

Question

我正在处理一个矩阵，我们称之为 X，在 python 中。

我知道如何使用 X.shape 获取矩阵的维度，但我对在 for 循环中使用矩阵的行数特别感兴趣，我不知道如何在适合的数据类型中获取此值一个循环。

例如，想象一个简单的情况：

a = np.matrix([[1,2,3],[4,5,6]])
for i in 1:(number of rows of a)
     print i

如何自动获得“a 的行数”？

Answer 1

X.shape[0] == X 中的行数

Answer 2

在numpy 上进行表面搜索将引导您到shape。它返回一个数组维度的元组。

在您的情况下，第一个维度（轴）涉及列。您可以像访问元组的元素一样访问它：

import numpy as np

a = np.matrix([[1,2,3],[4,5,6]])
# a. shape[1]: columns
for i in range(0,a.shape[1]):
   print 'column '+format(i)

# a. shape[0]: rows   
for i in range(0, a.shape[0]):
   print 'row '+format(i)

这将打印：

column 0
column 1
column 2
row 0
row 1