Python | 熊猫 DataFrame.get()

简介

在使用熊猫风格的数据框架（DataFrame）操作数据时，我们可能需要获取某一列中的特定值。虽然可以使用类似 dataframe[columnname][rowindex] 的方法获取值，但这个过程的可读性不佳。熊猫 DataFrame 已经提供了 get 方法来获取指定列的值。

下面是 get 方法的语法：

dataframe.get(key, default=None)

其中，key 是列的名称，default 是默认值（当指定的列不存在或索引超出范围时返回）。如果默认值为 None，则在出现这些情况时将引发 KeyError。

使用举例

以下是可以决定股票是否应该购买的简单程序的示例。

import pandas as pd

data = {'name': ['Apple Inc.', 'Microsoft Corp.', 'Amazon.com, Inc.'],
        'price': [129.87, 249.73, 3177.29],
        'market cap': ['$2.12 T', '$1.89 T', '$1.58 T']}

df = pd.DataFrame(data)

# Print stock data
print(df)

# Use get() method to retrieve the stock price of Apple Inc.
apple_price = df.get('price').iloc[0]
print(f"Stock price for Apple Inc. is {apple_price}.")

# Use get() method with default value to retrieve the stock price for an unknown company
unknown_price = df.get('XYZ', default='N/A')
print(f"Stock price for unknown company is {unknown_price}")

上面的程序创建了一个简单的数据框架，其中包含三个公司的数据。接下来，程序使用 get 方法检索苹果公司的股票价格，并打印出这个价格。

此外，该程序还示范了如何使用 get 方法来检索未知公司的股票价格。由于我们没有一家名为“XYZ”的公司的数据，我们设置了一个默认值 "N/A"。

总结

get 方法是熊猫 DataFrame 操作中非常有用的方法之一，它可以提高代码的可读性和可维护性。当我们需要检索特定列的值时，可以使用 get 方法来获取相关数据，而无需编写复杂的通过索引和列名访问数据的代码。