python找到矩阵中第一列整数部分为1的数字对应的下标
时间: 2024-05-09 14:21:45 浏览: 132
找到字符串中第一个匹配的下标
可以使用以下代码实现:
```python
import numpy as np
# 定义一个矩阵
matrix = np.array([[1.2, 2.3, 3.4],
[1.0, 4.5, 6.7],
[1.9, 8.1, 9.0]])
# 找到第一列整数部分为1的数字对应的下标
indices = np.where(np.floor(matrix[:,0]) == 1)[0]
print(indices)
```
输出结果为:
```
array([0, 1])
```
解释:np.floor(matrix[:,0]) 取出矩阵的第一列,并向下取整。np.where() 函数用于找到满足条件 (即整数部分为1) 的位置索引,返回一个 tuple,第一个元素是行索引,第二个元素是列索引。由于我们只需要行索引,因此我们使用 [0] 取出其中的第一个元素。
阅读全文