利用 Python 进行数据分析：原书第 3 版

NumPy

基础：数组和向量化计算

|

113

另一个函数 numpy.in1d 用于检查一个数组中的值是否在另一个数组中，并返回一个布

尔型数组：

NumPy

的数组集合函数参见表

4-7

。

表 4-7：数组集合操作

方法 说明

unique(x)

计算

x

中的唯一值

，

并返回有序结果

intersect1d(x, y)

计算

x

和

y

的交集

，

并返回有序结果

union1d(x, y)

计算

x

和

y

的并集

，

并返回有序结果

in1d(x, y)

计算

x

中的元素是否包含于

y，

返回布尔型数组

setdiff1d(x, y)

计算在

x

中且不在

y

中的元素差集

setxor1d(x, y)

计算存在于一个数组中但不同时存在于两个数组中的异或元素集

4.5

使用数组进行文件输入和输出

NumPy

能够读写磁盘上的文本数据或二进制格式的数据。本节只讨论

NumPy

内置的

二进制格式，因为更多的用户会使用

pandas

或其他工具加载文本或表格数据（详见第

6

章 ）。

numpy.save 和 numpy.load 是读写磁盘数组数据的两个主要函数。默认情况下，数组是

以未压缩的原始二进制格式保存在扩展名为

.npy

的文件中的：

如果文件路径末尾没有扩展名

.npy

，则该扩展名会自动加上。然后就可以通过 numpy.

load 读取磁盘上的数组了：

通过 numpy.savez 可以将多个数组保存到一个未压缩文件中，将数组以关键字参数的形

式传入即可：