Python | os.sched_getaffinity() 方法

简介

os.sched_getaffinity() 方法是Python内置的一个os模块函数，用于获取当前进程在CPU上的亲和性（affinity）配置。

在多核CPU系统中，操作系统会将进程分配到不同的CPU核心上执行。亲和性（affinity）是指进程与CPU核心之间的关联关系。通过亲和性配置，可以指定进程在哪些CPU核心上运行，以优化系统的性能。

语法

os.sched_getaffinity(pid)

参数

• pid（可选）：指定进程的PID（进程标识符）。如果未提供pid参数，则默认为当前进程的PID。

返回值

os.sched_getaffinity() 方法返回一个可迭代对象（set），其中包含当前进程在CPU上的亲和性配置。

示例

下面的示例展示了如何使用os.sched_getaffinity() 方法获取当前进程的亲和性配置：

import os

# 获取当前进程的亲和性配置
affinity = os.sched_getaffinity()

# 输出亲和性配置
print(f"Affinity: {affinity}")

输出结果如下：

Affinity: {1, 2, 3, 4}

以上示例表示当前进程被分配到CPU核心 1、2、3、4 上运行。

注意事项

1. 在某些操作系统上，可能需要 root 权限才能访问 os.sched_getaffinity() 方法。
2. 需要注意的是，os.sched_getaffinity() 方法的返回结果可能会因为不同的操作系统和版本而有所不同。因此，在使用此方法时应谨慎处理返回值。