計算 Pandas 中的滾動相關值
Preet Sanghavi
2022年5月16日
本教程將討論如何在 Pandas 中查詢滾動相關值。
在 Pandas 中計算滾動相關值的步驟
以下是計算 Pandas DataFrame 兩列之間滾動相關性的步驟。
進口 Pandas
我們需要從匯入 Pandas 庫開始。
import pandas as pd
建立一個 DataFrame
現在讓我們建立一個帶有兩列的示例 Pandas DataFrame,我們將在兩列之間計算滾動相關性。
data = {'Data1': [1, 4, 7, 10], 'Data2': [2, 5, 8, 11]}
df = pd.DataFrame(data)
我們建立了一個名為 data
的字典,其中包含 Data1
和 Data2
兩列,並將該字典傳遞給 pd.DataFrame()
函式以建立如下所示的 Pandas DataFrame。
print(df)
輸出:
Data1 Data2
0 1 2
1 4 5
2 7 8
3 10 11
計算滾動相關性
我們將使用 Pandas 中的 rolling()
函式滾動我們的第一列,然後使用 corr()
函式計算滾動列與 DataFrame 中另一列的相關性。
rc = df['Data1'].rolling(2).corr(df['Data2'])
我們通過兩個觀察值的視窗長度將第一列滾動 2 並將其與第二列相關聯。我們將相關值儲存在一個新變數中。
現在讓我們列印新變數以檢視兩列之間滾動相關性的值。
print(rc)
輸出:
0 NaN
1 1.0
2 1.0
3 1.0
上面的輸出顯示了 DataFrame 中兩列之間的滾動相關值。因此,我們可以使用上述技術成功地確定 Pandas 中兩個 DataFrame 列之間所需的滾動相關值。
Author: Preet Sanghavi