admin 管理员组

文章数量: 1184232


2024年3月9日发(作者:轻量级线程和进程的区别)

numpy基本操作

numpy是Python中重要的数学库之一,主要用于处理多维数组和矩阵运算。本文介绍numpy的一些基本操作,包括创建数组、数组运算、数组索引和切片、数组变形等。

1. 创建数组

可以通过numpy的array函数创建数组,传入一个列表或元组即可:

import numpy as np

a = ([1, 2, 3]) # 一维数组

print(a) # [1 2 3]

b = ([[1, 2, 3], [4, 5, 6]]) # 二维数组

print(b) # [[1 2 3]

# [4 5 6]]

还可以通过numpy的arange函数创建等差数组:

c = (0, 10, 2) # 从0开始,每隔2取一个,到10停止

print(c) # [0 2 4 6 8]

2. 数组运算

numpy数组支持各种运算,包括加减乘除、幂、三角函数等,这些运算都是按元素进行的。

a = ([1, 2, 3])

b = ([4, 5, 6])

- 1 -

c = a + b # 数组加法

print(c) # [5 7 9]

d = a * b # 数组乘法

print(d) # [ 4 10 18]

e = (a) # sin函数

print(e) # [0.84147098 0.90929743 0.14112001]

numpy还支持数组与标量的运算,比如将数组每个元素都加上2:

a = ([1, 2, 3])

b = a + 2

print(b) # [3 4 5]

3. 数组索引和切片

与Python列表类似,numpy数组也可以通过索引和切片来访问元素。

a = ([1, 2, 3])

print(a[0]) # 1

b = ([[1, 2, 3], [4, 5, 6]])

print(b[1, 2]) # 6

切片可以用[start:end:step]的形式表示,其中start是起始索引,end是结束索引(不包含),step是步长。

a = ([1, 2, 3, 4, 5])

print(a[1:4:2]) # [2 4]

b = ([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

- 2 -

print(b[:2, 1:]) # [[2 3]

# [5 6]]

4. 数组变形

numpy数组可以通过reshape函数改变形状:

a = ([1, 2, 3, 4, 5, 6])

b = e((2, 3))

print(b) # [[1 2 3]

# [4 5 6]]

可以使用ravel函数将多维数组转换为一维数组:

c = ()

print(c) # [1 2 3 4 5 6]

还可以使用transpose函数进行转置:

d = ose()

print(d) # [[1 4]

# [2 5]

# [3 6]]

以上就是numpy的一些基本操作,更多详细内容可以参考numpy官方文档。

- 3 -


本文标签: 数组 运算 函数 创建