将ndarray转化为列表
时间: 2023-08-08 21:12:27 浏览: 43
你可以使用numpy库中的tolist()函数将ndarray转换为列表。例如:
```
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
lst = arr.tolist()
print(lst) # Output: [1, 2, 3, 4, 5]
```
如果你的ndarray是多维的,tolist()函数也可以将其转换为嵌套列表。例如:
```
import numpy as np
arr = np.array([[1, 2], [3, 4], [5, 6]])
lst = arr.tolist()
print(lst) # Output: [[1, 2], [3, 4], [5, 6]]
```
相关问题
将ndarray转化为tensor
在PyTorch中,可以使用`torch.tensor()`函数将numpy ndarray转换为tensor。例如:
```python
import torch
import numpy as np
ndarray = np.array([[1, 2], [3, 4]])
tensor = torch.tensor(ndarray)
print(tensor)
```
输出结果为:
```
tensor([[1, 2],
[3, 4]])
```
注意,使用`torch.tensor()`函数转换时,会创建一个新的tensor,该tensor与原始ndarray共享底层数据存储,但是具有不同的属性(如数据类型、形状等)。如果需要将ndarray转换为与原tensor具有相同属性的新tensor,可以使用`torch.from_numpy()`函数。例如:
```python
import torch
import numpy as np
tensor = torch.tensor([[1, 2], [3, 4]])
ndarray = tensor.numpy()
new_tensor = torch.from_numpy(ndarray)
print(new_tensor)
```
输出结果为:
```
tensor([[1, 2],
[3, 4]])
```
将ndarray转化为整数型
可以使用astype方法将ndarray转换为整数型,具体实现代码如下:
```python
import numpy as np
arr = np.array([1.2, 2.5, 3.8])
int_arr = arr.astype(int)
print(int_arr)
```
输出:
```
[1 2 3]
```
在这个例子中,我们创建了一个包含浮点数的ndarray,然后使用astype方法将其转换为整数型。注意,浮点数会被截断成整数。