Python获取环境变量|极客教程

Python获取环境变量

在编写Python程序时，有时候我们需要获取系统的环境变量，以便根据不同的环境设置不同的操作。Python提供了一个简单而方便的方法来访问和操作环境变量，让我们来详细了解一下。

环境变量是一种在操作系统中用来传递信息给正在运行的进程的方法。它们通常用来配置不同程序的行为，例如指示程序在哪里查找可执行文件、哪个用户执行程序等。在Unix/Linux系统中，环境变量通常存储在Shell的session环境中，而在Windows系统中，环境变量保存在注册表中。

Python提供了一个名为os的标准库，用于与操作系统进行交互。通过os模块，我们可以轻松地访问和操作环境变量。下面是一些常用的方法：

我们可以使用os.environ来获取所有环境变量的字典，其中键是环境变量的名称，值是环境变量的值。下面是一个简单的示例代码：

import os

env_vars = os.environ
for key, value in env_vars.items():
    print(f"{key}: {value}")

运行以上代码，将打印出所有的环境变量及其对应的值。在不同的操作系统上可能会有不同的环境变量，你可以尝试在不同系统上运行该代码来查看不同的结果。

如果我们只想获取特定的环境变量，可以使用os.getenv方法，传入环境变量的名称即可。如果该环境变量不存在，则可以设置默认值，如下所示：

import os

python_path = os.getenv("PYTHONPATH", default="No Python Path")
print(python_path)

在这个示例中，我们尝试获取名为PYTHONPATH的环境变量，如果该环境变量不存在，则打印出No Python Path。你可以尝试修改环境变量的名称，看看会输出什么结果。

有时候我们也需要在Python程序中设置环境变量，这可以通过os.environ字典来实现。只需将新的环境变量键值对添加到os.environ中即可。下面是一个示例代码：

import os

os.environ["NEW_VARIABLE"] = "New Value"
new_var = os.getenv("NEW_VARIABLE")
print(new_var)

在这个示例中，我们将一个新的环境变量NEW_VARIABLE设置为New Value，然后再次获取该环境变量并输出其值。你可以尝试将以上代码复制到你的Python解释器中运行，看看是否能成功设置并获取新的环境变量。

通过Python获取和操作环境变量是一个非常有用的功能，在实际编程中经常会遇到。在本文中，我们学习了如何使用Python的os模块来获取所有环境变量、获取特定环境变量、以及设置新的环境变量。