numpy.ndarray如何知道某个数据的索引位置
时间: 2023-05-21 18:06:39 浏览: 63
你可以通过numpy.where()方法来查找给定元素在数组中的索引位置。例如,如果你有一个numpy的二维数组x,你可以使用以下代码来查找元素1的位置:
import numpy as np
x = np.array([[0, 1, 2], [3, 4, 5]])
index = np.where(x == 1)
print(index)
输出结果将是:
(array([0]), array([1]))
这意味着元素1位于数组x的第0行第1列(也就是第二个元素)。
相关问题
numpy.ndarray中删除数据
可以使用numpy.delete()函数来删除numpy.ndarray中的数据。该函数的语法如下:
```python
numpy.delete(arr, obj, axis=None)
```
其中,arr是输入的numpy.ndarray数组;obj是要删除的元素的索引或者切片;axis是要删除的轴的索引。
例如,我们可以删除一个一维数组中的第三个元素,代码如下:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
new_arr = np.delete(arr, 2)
print(new_arr) # [1 2 4 5]
```
我们也可以删除一个二维数组中的第二行,代码如下:
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
new_arr = np.delete(arr, 1, axis=0)
print(new_arr)
# [[1 2 3]
# [7 8 9]]
```
需要注意的是,numpy.delete()函数并不会修改原始的numpy.ndarray数组,而是返回一个新的数组。
将numpy.ndarray类型的数据写入excel表
可以使用`pandas`库将`numpy.ndarray`类型的数据写入Excel表格。下面是一个示例代码:
```python
import pandas as pd
import numpy as np
# 创建一个numpy.ndarray类型的数据
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 将数据转换为pandas的DataFrame
df = pd.DataFrame(data)
# 将DataFrame写入Excel表格
df.to_excel('output.xlsx', index=False)
```
在这个示例中,我们首先创建了一个`numpy.ndarray`类型的数据,然后将其转换为`pandas`的`DataFrame`。最后,使用`to_excel`函数将DataFrame写入名为"output.xlsx"的Excel表格中。`index=False`表示不将索引列写入Excel文件。
你可以根据自己的需求修改代码,例如自定义列名、写入特定的工作表等。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)