numpy基础（创建数组，类型转换，通用函数）

呀比小饼干

已于 2022-03-20 01:09:02 修改

阅读量897

点赞数

分类专栏： StudyPython 文章标签： python 机器学习数据挖掘

于 2022-03-18 12:24:13 首次发布

本文链接：https://blog.csdn.net/qq_45850131/article/details/123571416

版权

12 篇文章 1 订阅

订阅专栏

创建数组

# 使用array函数创建数组
import numpy as np
data = [[1, 2, 3, 4, 5],[6, 7, 8, 9, 10]]
arr1 = np.array(data) 
arr1

在这里插入图片描述

# 查看维度
arr1.ndim
# ouput:2

# 使用arange创建数组
arr2 = np.arange(1,10)
arr2

# ones()函数创建全1数组
arr3 = np.ones((2, 3, 2))

在这里插入图片描述

# zeros()函数创建全0数组
arr4 = np.zeros((3, 2))
arr4

# empty()函数创建空数组
arr5 = np.empty((2, 3))
arr5

# 查看数据类型
arr = np.ones(10)
arr.dtype
# out:dtype('float64')

# 设置数据类型
arr = np.array([1, 2, 3, 4, 5], dtype=np.string_)
arr.dtype

在这里插入图片描述

# 类型转换 astype
arr = np.arange(10)
arr.dtype  # dtype('int32')
float_arr = arr.astype(np.float64)
float_arr.dtype  # dtype('float64')

# numpy转tensor
arr = torch.from_numpy(arr)

在这里插入图片描述

# tensor转numpy
arr = arr.numpy()

在这里插入图片描述

type(weekday)
# out:numpy.ndarray
weekday = weekday.tolist()
type(weekday)
# out:list

type(weekday)
# out:list
weekday = np.array(weekday)
type(weekday)
# out:numpy.ndarray

在这里插入图片描述

在这里插入图片描述

关注