Pandas:使用np.where()删除值小于今天的行？

基础概念

pandas 是一个强大的数据处理和分析库，而 numpy 是一个用于科学计算的库。np.where() 是 numpy 中的一个函数，用于根据条件返回输入数组中的元素。

类型

np.where() 函数的基本语法如下：

numpy.where(condition, x, y)

condition：一个布尔数组，用于指定条件。
x：当条件为 True 时返回的值。
y：当条件为 False 时返回的值。

应用场景

在数据处理中，np.where() 常用于根据某些条件过滤或替换数据。

示例代码

假设我们有一个包含日期数据的 pandas DataFrame，并且我们希望删除日期小于今天的行。以下是一个示例代码：

import pandas as pd
import numpy as np
from datetime import datetime

# 创建示例 DataFrame
data = {
    'date': ['2023-01-01', '2023-02-01', '2023-03-01', '2023-04-01'],
    'value': [10, 20, 30, 40]
}
df = pd.DataFrame(data)

# 将日期列转换为 datetime 类型
df['date'] = pd.to_datetime(df['date'])

# 获取今天的日期
today = datetime.today()

# 使用 np.where() 删除日期小于今天的行
df = df[np.where(df['date'] >= today, True, False)]

print(df)

解释

创建示例 DataFrame：我们首先创建一个包含日期和值的 DataFrame。
转换日期类型：将日期列转换为 datetime 类型，以便进行日期比较。
获取今天的日期：使用 datetime.today() 获取当前日期。
使用 np.where() 过滤数据：np.where(df['date'] >= today, True, False) 生成一个布尔数组，表示哪些行的日期大于等于今天。然后我们使用这个布尔数组过滤 DataFrame。