Python Pandas 中的 shift() 函数,实现数据平移的完美利器!

Python Pandas 中的 shift() 函数,实现数据平移的完美利器!_第1张图片

更多Python学习内容:ipengtao.com

shift() 是 Pandas 中一个常用的数据处理函数,它用于对数据进行移动或偏移操作,常用于时间序列数据或需要计算前后差值的情况。本文将详细介绍 shift() 函数的用法,包括语法、参数、示例以及常见应用场景。

什么是 shift() 函数?

shift() 函数是 Pandas 库中的一个数据处理函数,用于将数据按指定方向移动或偏移。它可以对时间序列数据或其他类型的数据进行操作,通常用于计算时间序列数据的差值、百分比变化等。该函数的主要作用是将数据移动到指定的行或列,留下空白或填充 NaN 值。

shift() 函数的语法

shift() 函数的基本语法如下:

DataFrame.shift(periods=1, freq=None, axis=0, fill_value=None)

参数说明:

  • periods:指定移动的步数,可以为正数(向下移动)或负数(向上移动)。默认为 1。

  • freq:可选参数,用于指定时间序列数据的频率,通常用于时间序列数据的移动操作。

  • axis:指定移动的方向,可以为 0(默认,沿行移动)或 1(沿列移动)。

  • fill_value:可选参数,用于填充移动后留下的空白位置,通常为填充 NaN 值。

shift() 函数的示例

通过一些示例来演示 shift() 函数的用法。

示例 1:向下移动数据

import pandas as pd

data = {'A': [1, 2, 3, 4, 5],
        'B': [10, 20, 30, 40, 50]}
df = pd.DataFrame(data)

# 向下移动一行数据
df_shifted = df.shift(periods=1)
print(df_shifted)

输出结果:

A     B
0  NaN   NaN
1  1.0  10.0
2  2.0  20.0
3  3.0  30.0
4  4.0  40.0

在这个示例中,创建了一个包含两列数据的 DataFrame,并使用 shift() 函数向下移动了一行数据。移动后,第一行的数据被填充为 NaN。

示例 2:向上移动数据

import pandas as pd

data = {'A': [1, 2, 3, 4, 5],
        'B': [10, 20, 30, 40, 50]}
df = pd.DataFrame(data)

# 向上移动一行数据
df_shifted = df.shift(periods=-1)
print(df_shifted)

输出结果:

A     B
0  2.0  20.0
1  3.0  30.0
2  4.0  40.0
3  5.0  50.0
4  NaN   NaN

这个示例,使用负数的 periods 参数将数据向上移动了一行。最后一行的数据被填充为 NaN。

示例 3:向右移动列数据

import pandas as pd

data = {'A': [1, 2, 3, 4, 5],
        'B': [10, 20, 30, 40, 50]}
df = pd.DataFrame(data)

# 向右移动一列数据
df_shifted = df.shift(periods=1, axis=1)
print(df_shifted)

输出结果:

A     B
0  NaN   1.0
1  NaN   2.0
2  NaN   3.0
3  NaN   4.0
4  NaN   5.0

在这个示例中,使用 axis=1 参数将列数据向右移动了一列,左边填充为 NaN。

示例 4:指定填充值

import pandas as pd

data = {'A': [1, 

2, 3, 4, 5],
        'B': [10, 20, 30, 40, 50]}
df = pd.DataFrame(data)

# 向下移动一行数据,填充空白处为 0
df_shifted = df.shift(periods=1, fill_value=0)
print(df_shifted)

输出结果:

A   B
0  0   0
1  1  10
2  2  20
3  3  30
4  4  40

在这个示例中,使用 fill_value 参数指定了填充值为 0,因此移动后的空白位置被填充为 0。

常见应用场景

shift() 函数在处理时间序列数据、计算数据差值、计算百分比变化等方面非常有用。

1. 计算时间序列数据的差值

import pandas as pd

# 创建一个包含时间序列数据的DataFrame
data = {'Date': pd.date_range(start='2023-01-01', periods=5, freq='D'),
        'Price': [100, 105, 110, 108, 112]}
df = pd.DataFrame(data)

# 计算每日价格的差值
df['Price_Diff'] = df['Price'].diff()
print(df)

输出结果:

Date  Price  Price_Diff
0 2023-01-01    100         NaN
1 2023-01-02    105         5.0
2 2023-01-03    110         5.0
3 2023-01-04    108        -2.0
4 2023-01-05    112         4.0

2. 计算数据的滞后值或前值

import pandas as pd

# 创建一个包含数据的DataFrame
data = {'Value': [10, 20, 30, 40, 50]}
df = pd.DataFrame(data)

# 计算数据的滞后值(前一行的值)
df['Lagged_Value'] = df['Value'].shift(1)
print(df)

输出结果:

Value  Lagged_Value
0     10           NaN
1     20          10.0
2     30          20.0
3     40          30.0
4     50          40.0

3. 计算数据的百分比变化

import pandas as pd

# 创建一个包含数据的DataFrame
data = {'Sales': [1000, 1200, 1500, 1300, 1600]}
df = pd.DataFrame(data)

# 计算数据的百分比变化
df['Percentage_Change'] = (df['Sales'] - df['Sales'].shift(1)) / df['Sales'].shift(1) * 100
print(df)

输出结果:

Sales  Percentage_Change
0   1000                 NaN
1   1200                20.0
2   1500                25.0
3   1300               -13.3
4   1600                23.1

4. 创建滑动窗口统计信息

import pandas as pd

# 创建一个包含数据的DataFrame
data = {'Value': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}
df = pd.DataFrame(data)

# 计算数据的滑动平均值(窗口大小为3)
df['Moving_Average'] = df['Value'].rolling(window=3).mean()
print(df)

输出结果:

Value  Moving_Average
0      1              NaN
1      2              NaN
2      3              2.0
3      4              3.0
4      5              4.0
5      6              5.0
6      7              6.0
7      8              7.0
8      9              8.0
9     10              9.0

总结

shift() 函数是 Pandas 中用于移动或偏移数据的重要工具。它可以处理时间序列数据、计算数据差值以及进行数据预处理。通过本文的介绍和示例,应该已经掌握了 shift() 函数的基本用法和常见应用场景。在实际数据分析和处理中,熟练使用这个函数将有助于提高工作效率和数据处理的精度。

如果你觉得文章还不错,请大家 点赞、分享、留言 下,因为这将是我持续输出更多优质文章的最强动力!

更多Python学习内容:ipengtao.com

干货笔记整理

  100个爬虫常见问题.pdf ,太全了!

Python 自动化运维 100个常见问题.pdf

Python Web 开发常见的100个问题.pdf

124个Python案例,完整源代码!

PYTHON 3.10中文版官方文档

耗时三个月整理的《Python之路2.0.pdf》开放下载

最经典的编程教材《Think Python》开源中文版.PDF下载

Python Pandas 中的 shift() 函数,实现数据平移的完美利器!_第2张图片

点击“阅读原文”,获取更多学习内容

你可能感兴趣的:(Python,python,pandas,数据库,开发语言)