Python NumPy教程之数组的基本操作详解

本教程将详细讲解在Python NumPy中如何进行数组的基本操作，本文主要内容包括：

• 数组创建
• 数组的维度
• 数组变形
• 数组切片
• 数组索引

数组创建

创建数组是使用NumPy的第一步。在NumPy中，数组可以通过多种方式创建。以下是一些常见的方式：

1. 使用array函数创建数组

最简单的创建数组的方法是使用array函数。将Python列表传递给array函数，可以将列表转换为NumPy数组：

import numpy as np

# 创建数组
a = np.array([1, 2, 3])

# 打印数组
print(a)

输出结果：

[1 2 3]

2.使用zeros或ones函数创建数组

zeros和ones函数分别用于创建特定形状的数组，并将所有项初始化为0或1。

import numpy as np

# 创建长度为3的全零数组
a = np.zeros(3)
print(a)

# 创建长度为3的全一数组
b = np.ones(3)
print(b)

# 创建形状为(2, 3)的全零数组
c = np.zeros((2, 3))
print(c)

# 创建形状为(2, 3)的全一数组
d = np.ones((2, 3))
print(d)

输出结果：

[0. 0. 0.]
[1. 1. 1.]
[[0. 0. 0.]
 [0. 0. 0.]]
[[1. 1. 1.]
 [1. 1. 1.]]

数组的维度

在NumPy中，数组的维度指的是数组的形状。可以使用shape属性查看数组的形状。

import numpy as np

# 创建numpy数组
a = np.zeros((2, 3))

# 打印数组形状
print(a.shape)

输出结果：

(2, 3)

数组变形

在NumPy中，可以使用reshape函数更改数组的形状。

import numpy as np

# 创建numpy数组
a = np.zeros((2, 3))

# 变形数组
b = a.reshape((3, 2))

# 打印数组形状
print(b.shape)

输出结果：

(3, 2)

数组切片

在NumPy中，可以使用切片从数组中抽取一个子集。

import numpy as np

# 创建numpy数组
a = np.array([1, 2, 3, 4, 5])

# 切片数组
b = a[1:3]

# 打印数组
print(b)

输出结果：

[2 3]

数组索引

在NumPy中，可以使用索引访问数组中的元素。

import numpy as np

# 创建numpy数组
a = np.array([1, 2, 3])

# 访问数组中的元素
print(a[0])

输出结果：

1

以上便是Python NumPy数组基本操作的详细攻略，希望对您的学习有所帮助。

示例1：使用zeros或ones函数创建数组，创建3维数组

import numpy as np

# 创建形状为(2, 2, 2)的全零数组
a = np.zeros((2, 2, 2))
print(a)

# 创建形状为(2, 2, 2)的全一数组
b = np.ones((2, 2, 2))
print(b)

输出结果：

[[[0. 0.]
  [0. 0.]]

 [[0. 0.]
  [0. 0.]]]
[[[1. 1.]
  [1. 1.]]

 [[1. 1.]
  [1. 1.]]]

示例2：数组变形，将6x1的一维数组转化为3x2的二维数组

import numpy as np

# 创建numpy数组
a = np.array([1, 2, 3, 4, 5, 6])

# 变形数组
b = a.reshape((3, 2))

# 打印数组形状
print(b.shape)
# 打印数组
print(b)

输出结果：

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