python中list,ndarray,Tensor间的转换小结

更新时间：2024年02月02日 11:30:58 作者：tao355667

数据类型转换是常见的功能,本文主要介绍了python中list,ndarray,Tensor间的转换小结,文中通过示例代码介绍的非常详细,需要的朋友们下面随着小编来一起学习学习吧

一、list,ndarray,Tensor间的转化

废话不多说，看表格就行

数据类型	所属包
list	python
ndarray	numpy
Tensor	pytorch

转化类型	对应API	注意点
list转换为ndarray	`numpy.array()`
ndarray转换为list	`ndarray对象.tolist()`
list转换为Tensor	`torch.tensor()`	list中的int类型数据转换后会变为folat，若需要保持int，则转换时需要加上类型
Tensor转换为list	`Tensor对象.tolist()`
ndarray转换为Tensor	`torch.from_numpy()torch.tensor()`
Tensor(CPU)转换为ndarray	`Tensor对象.numpy()`	GPU上的Tensor不能直接转换为numpy，需要间接转换
Tensor(GPU)转换为ndarray	`Tensor对象.cpu().numpy()`	GPU上的Tensor不能直接转换为numpy，需要间接转换

ndarray --> PILimage

From PIL import Image
y = Image.fromarray(array)

PILimage --> ndarray

From PIL import Image
image = Image.open(“…..”)
img = np.asarray(image)

Tensor --> ndarray

import numpy as np
yy = np.array(tensor)

ndarray --> Tensor

tensor = torch.from_numpy(ndarray)

tip：返回的张量和ndarray共享同一内存。对张量的修改将反映在ndarray中，反之亦然。

ndarray数据转换数据类型

array.astype(np.uint8)

将array复制，并将数据类型强制转化为int8

ndarray --> List

import numpy as np
n = np.array([[1,2],[3,4],[5,6]])
m = n.tolist()

List --> Tensor

tensor = torch.tensor(list)

二、例程

import numpy as np
import torch

#list转换为ndarray
li=[[1,2,3],[4,5,6],[7,8,9]]
a=np.array(li)  #list转换为ndarray
print(a) 
print(type(a),'\n')

#ndarray转换为list
b=a.tolist()#ndarray转换为list
print(b) 
print(type(b),'\n')

#list转换为Tensor
li=[[1,2,3],[4,5,6],[7,8,9]]
c=torch.tensor(li)  #list转换为Tensor
print(c)
print(type(c),'\n')

#Tensor转换为list
d=c.tolist() #Tensor转换为list
print(d) 
print(type(d),'\n')

#ndarray转换为Tensor
nd=np.arange(0,12).reshape(3,4)
e=torch.from_numpy(nd)  #ndarray转换为Tensor
# e=torch.tensor(nd)    #ndarray转换为Tensor
print(e) 
print(type(e),'\n')

#Tensor转换为ndarray
f=e.numpy()
print(f) 
print(type(f),'\n')

运行结果

[[1 2 3]
[4 5 6]
[7 8 9]]
<class 'numpy.ndarray'>

[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
<class 'list'>

tensor([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
<class 'torch.Tensor'>

[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
<class 'list'>

tensor([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]], dtype=torch.int32)
<class 'torch.Tensor'>

[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]
<class 'numpy.ndarray'>

到此这篇关于python中list,ndarray,Tensor间的转化小结的文章就介绍到这了,更多相关python list,ndarray,Tensor转化内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Python如何新建三维数组并赋值
本文详细介绍了如何使用Python和numpy库建立三维数组并对其进行赋值。首先，通过numpy创建一个3x3x3的三维数组，其次，将自定义的二维数组赋值到三维数组中。本文还解释了相关参数的含义，使读者能够更好地理解和应用到其他多维矩阵的操作中
2024-09-09
Python实现中文文本关键词抽取的三种方法
文本关键词抽取,是对文本信息进行高度凝练的一种有效手段,通过3-5个词语准确概括文本的主题,帮助读者快速理解文本信息,本文分别采用TF-IDF方法、TextRank方法和Word2Vec词聚类方法,利用Python语言进行开发,实现文本关键词的抽取,需要的朋友可以参考下
2024-01-01
Python解析json时提示“string indices must be integers”问题解决方法
这篇文章主要介绍了Python解析json时提示“string indices must be integers”问题解决方法,结合实例形式分析了Python解析json字符串操作规范与相关使用技巧,需要的朋友可以参考下
2019-07-07
Python实现视频下载功能
最近一两年短视频业务风生水起，各个视频网站都有各自特色的短视频内容。如果有一个程序可以把各大视频网站的热门用户最新发布的视频下载下来，不仅方便了观看，还可以将没有版权的视频发布在个人社交网站上，增加自己的人气，多好呀
2017-03-03
python文件夹分区的实现方法实例
这篇文章主要给大家介绍了关于python文件夹分区的实现方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-04-04
python 用递归实现通用爬虫解析器
这篇文章主要介绍了python 用递归实现通用爬虫解析器的方法，帮助大家更好的理解和学习使用python，感兴趣的朋友可以了解下
2021-04-04
10行Python代码计算汽车数量的实现方法
这篇文章主要介绍了10行Python代码计算汽车数量的实现方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-10-10
pygame游戏之旅载入小车图片、更新窗口
这篇文章主要为大家详细介绍了pygame游戏之旅的第3篇，教大家如何载入小车图片、更新窗口，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-11-11
Python编程中被忽视的核心技巧总结
这篇文章主要介绍了一些在 Python 编程中可能被忽视的核心功能，包括默认参数、海象运算符、*args 和 **kwargs 的使用等，需要的可参考下
2023-08-08
django自带的权限管理Permission用法说明
这篇文章主要介绍了django自带的权限管理Permission用法说明，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-05-05