如何在循环

时间:2016-04-20 17:28:21

标签: python-2.7 numpy

我正在使用矩阵,让我们在python中调用它。

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

例如,想象一下简单的情况:

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

如何自动获得“行数”?

2 个答案:

答案 0 :(得分:2)

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

答案 1 :(得分:1)

numpy上的表面搜索会引导您shape。它返回数组维度的元组

在您的情况下,第一个维度(ax)涉及列。您可以在访问元组元素时访问它:

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