数组创建

NumPy 教程 · 第 3 章 · 6 次浏览

数组有形状、维度、大小等属性;索引和切片取数据。多维数组的切片是数据分析高频操作。

属性

shape 形状、ndim 维度、size 元素总数、dtype 类型、T 转置。

索引切片

arr[0] 第一行;arr[:, 0] 第一列;arr[1:3] 行切片;布尔索引 arr[arr > 3] 条件筛选。

代码示例

import numpy as np

mat = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 属性
print("形状:", mat.shape)      # (3, 3)
print("维度:", mat.ndim)       # 2
print("总数:", mat.size)       # 9

# 索引
print(mat[0])                  # 第一行 [1 2 3]
print(mat[1][2])               # 6
print(mat[2, 0])               # 7

# 切片
print(mat[:, 1])               # 第二列 [2 5 8]
print(mat[0:2, 0:2])           # 左上 2x2

# 布尔索引
arr = np.array([1, 5, 3, 8, 2])
print(arr[arr > 3])            # [5 8]

# 条件替换
arr[arr > 3] = 0
print(arr)                     # [1 0 3 0 2]
📚 Python 语法速查
常用条目速查,完整版见对应教程章节。可复制代码到在线运行中测试。
写法 / 语法作用
print("hello")输出到控制台
name = "张三"变量赋值(无需声明类型)
if x > 0: ...条件判断(注意冒号和缩进)
for i in range(10): ...循环(缩进是语法的一部分)
while x < 10: ...条件循环
def fn(a, b): return a + b定义函数
class Person: ...定义类
list = [1, 2, 3]列表(可改)
tuple = (1, 2)元组(不可改)
dict = {"key": "value"}字典(键值对)
set = {1, 2, 3}集合(去重)
len(obj)取长度
str(x) / int(x) / float(x)类型转换
s.split(",")字符串按分隔符拆分
" ".join(list)列表拼接成字符串
import os导入模块
from math import sqrt从模块导入指定函数
try: ... except Exception as e: ...异常捕获
with open("a.txt", "r") as f: ...文件读取(自动关闭)
f"你好 {name}"f-string 格式化
lambda x: x * 2匿名函数
list(map(fn, arr))函数式处理列表
range(start, stop, step)生成数字序列
if __name__ == "__main__":主入口判断
pip install 包名安装第三方包(命令行)