首页 > 解决方案 > 用 Python 在矩阵中取元素

问题描述

import numpy as np

m = np.matrix('[1, 2; 3, 4]')
print(m[0][0])

我期待看到1。但它告诉我[[1 2]]。我怎样才能得到1呢?

标签: pythonnumpymatrix

解决方案


正如Tranbi提到的,np.matrix官方文档指出:

不再推荐使用这个类,即使是线性代数。而是使用常规数组。将来可能会删除该类。

如果你还想使用np.matrix,它支持np.matrix.item()方法,你可以通过以下方式获取第一个元素:

m.item((0, 0))

或者

m.item(0)

这将产生:

1

如果您确实遵循切换到np.array的建议,

m = np.array([[1, 2], [3, 4]])
m[0][0]

会产生1


推荐阅读