numpy基础

Posted 2020-11-28 wt7018

一、基础知识

1、安装

conda install numpy

2、什么是numpy?
Python中做科学计算的基础库，重在数值计算

二、创建数组

import numpy as np
# 方式一
np.array([1, 2, 3, 4, 5])
# 方式二
np.array(range(5))
# 方式三
np.arange(5)

注意：数组的类型

<class ‘numpy.ndarray‘>

三、数据类型

1、查看数据类型

arr3 = np.arange(10)
print(arr3.dtype)

2、指定数据类型

np.array(dtype=数据类型)

例子

arr2 = np.array(range(10), dtype=float)
print(arr2.dtype)

arr2 = np.array(range(10), dtype=bool)
print(arr2.dtype)
print(arr2)

3、修改数据类型

new_arr = arr.astype(类型)

注意：修改数据类型要重新赋值

# 指定数据类型为int64
arr2 = np.array(range(10), dtype=np.int64)
print(arr2.dtype)
print(arr2)
print("=" * 30)
# 修改数据类型为bool
new_arr2 = arr2.astype(bool)
print(new_arr2.dtype)
print(new_arr2)

4、保留固定位数的小数

new_arr = np.round(保留位数)

注意：保留固定位数的小数，是生成新的数组，需要重新赋值

例子

import numpy as np
import random

arr1 = np.array([random.random() for i in range(10)])
arr2 = np.round(arr1, 2)
print(arr2)