0
点赞
收藏
分享

微信扫一扫

ndarray和list的互相转换


我们知道python非常适合做机器学习方面的工作,究其原因是因为它对于矩阵,数组操作的便捷性。在python中​​list​​​列表对象和​​numpy​​​模块中的​​ndarray​​类型也可以很好的发生转换,但是在转换的过程中还是会或多或少地遇到一些特殊的情况,这里做一下简单的介绍。

import numpy as np


class listAndNdarray:
def __init__(self):
self.x = x = [[1],[2, 3]]
self.y = np.array(x)
print(self.y) # [list([1]) list([2, 3])]
print(type(self.y)) # <class 'numpy.ndarray'>
print(type(self.y[0])) # <class 'list'>


main = listAndNdarray()

这里我们可以看到当使用​​np.array​​​对​​list​​​对象进行类型转换后,我们进一步测试此时的​​self.y​​​为​​<class 'numpy.ndarray'>​​​类型,但是内层的第一部分仍旧为​​list​​,这是因为数组在创建的时候必须保证每一行和每一列元素的数量是相同的, 而这里的​​self.x​​​中第一部分只包含一个元素,而第二部分包含两个元素,因此​​np.array​​​无法完成对其整体的类型转换而依旧保持内层的类型。
我们可以看如下代码:

import numpy as np


class listAndNdarray:
def __init__(self):
self.x = x = [[1, 1],[2, 3]]
self.y = np.array(x)
print(self.y)
print(type(self.y)) # <class 'numpy.ndarray'>
print(type(self.y[0])) # <class 'numpy.ndarray'>


main = listAndNdarray()

这时我们可以看到,无论是外层还是内层,类型均已被转换为了​​<class 'numpy.ndarray'>​​。

那么​​ndarray​​类型是否可以向list类型转换呢?是可以的,代码如下:

import numpy as np


class ndarrayToList:
def __init__(self):
self.array = np.ones((2, 3))
self.list = self.array.tolist()
print(type(self.list)) # <class 'list'>


main = ndarrayToList()

我们可以看到使用​​tolist()​​​可以将​​ndarray​​​类型转换为​​list​​​类型。
由此我们可以得出,在科学计算中​​​list​​​与​​ndarray​​​在某种程度上是等价的。


举报

相关推荐

0 条评论