用python使用numpy创建下列数组[1,1,2,3,4,5,6,1,1,6],找出数组中1元素的位置索引并输出索引（使用两种方法）。的代码

方法1：

import numpy as np

# 创建数组
arr = np.array([1, 1, 2, 3, 4, 5, 6, 1, 1, 6])

# 使用 np.where 查找索引
indexes = np.where(arr == 1)[0]

# 输出索引
print("1元素的位置索引为：", indexes)

方法2：

import numpy as np

# 创建数组
arr = np.array([1, 1, 2, 3, 4, 5, 6, 1, 1, 6])

# 使用循环查找索引
indexes = []
for i in range(len(arr)):
    if arr[i] == 1:
        indexes.append(i)

# 输出索引
print("1元素的位置索引为：", indexes)

两种方法都可以得到相同的结果：[0, 1, 7, 8]。其中 `np.where` 函数返回的是一个元组，第一个元素是一个一维数组，表示符合条件的元素的下标；第二个元素是一个空元组，表示这个数组只有一个维度。因此我们通过 `[0]` 取出了第一个元素，即符合条件的元素的下标。使用循环查找索引的方法比较简单，但是效率可能不如 `np.where` 函数高。