创建一个长度为n的一维全为0的ndarray对象,然后让第m个元素等于1 输入n,m并用空格隔开, 且n>=m 【样例输入】 10 5 【样例输出】 [[0. 0. 0. 0. 1. 0. 0. 0. 0. 0.]]
时间: 2024-05-11 07:16:43 浏览: 83
好的,您的问题已经收到。以下是代码实现:
```
import numpy as np
n, m = map(int, input().split())
arr = np.zeros(n)
arr[m-1] = 1
arr = np.expand_dims(arr, axis=0)
print(arr)
```
如果需要创建长度为n的一维全为1的ndarray对象,可以将第m个元素赋值为0,然后取反即可,如下:
```
import numpy as np
n, m = map(int, input().split())
arr = np.ones(n)
arr[m-1] = 0
arr = np.expand_dims(arr, axis=0)
print(arr)
```
如有疑问,欢迎继续提问。
阅读全文