我已經看到了基于切片的更新熊貓資料框- 但我無法完全找到我的用例的答案。
考慮這個代碼,我有一個帶有“通道”和“值”列的起始表:
import sys
if sys.version_info[0] < 3:
from StringIO import StringIO
else:
from io import StringIO
import pandas as pd
TESTDATA = StringIO("""channel,value
A,10
A,11
A,12
A,13
B,20
B,22
B,24
B,26
B,28
C,100
C,105
C,110
C,115
C,120
C,125
C,130
""")
mychans = ["A", "B", "C"]
df = pd.read_csv(TESTDATA)
df.insert (2, "value_rel", df["value"] - df["value"][0])
print("Starting:")
print(df.head())
for tchan in mychans:
this_ch_data = df[df["channel"]==tchan]
df.loc[this_ch_data.index, "value_rel"] = this_ch_data["value"] - this_ch_data["value"][0]
最后,我想獲得帶有附加“value_rel”列的同一個表,該列將顯示相對于該通道(切片)中的第一個值的值;那是:
A, 10, 0
A, 11, 1
A, 12, 2
A, 13, 3
B, 20, 0
B, 22, 2
B, 24, 4
B, 26, 6
B, 28, 8
C,100, 0
C,105, 5
...
如果我只是this_ch_data["value_rel"] = this_ch_data["value"] - this_ch_data["value"][0]在for回圈中使用,我會得到“一個值正在嘗試在來自 DataFrame 的切片的副本上設定”,這是有道理的。
但是,當運行代碼時,我得到:
$ python3 test1.py
Starting:
channel value value_rel
0 A 10 0
1 A 11 1
2 A 12 2
3 A 13 3
4 B 20 10
Traceback (most recent call last):
File "C:/msys64/mingw64/lib/python3.9/site-packages/pandas/core/indexes/base.py", line 3361, in get_loc
return self._engine.get_loc(casted_key)
File "pandas/_libs/index.pyx", line 76, in pandas._libs.index.IndexEngine.get_loc
File "pandas/_libs/index.pyx", line 108, in pandas._libs.index.IndexEngine.get_loc
File "pandas/_libs/hashtable_class_helper.pxi", line 2131, in pandas._libs.hashtable.Int64HashTable.get_item
File "pandas/_libs/hashtable_class_helper.pxi", line 2140, in pandas._libs.hashtable.Int64HashTable.get_item
KeyError: 0
The above exception was the direct cause of the following exception:
Traceback (most recent call last):
File "C:\msys64\tmp\test1.py", line 38, in <module>
df.loc[this_ch_data.index, "value_rel"] = this_ch_data["value"] - this_ch_data["value"][0]
File "C:/msys64/mingw64/lib/python3.9/site-packages/pandas/core/series.py", line 942, in __getitem__
return self._get_value(key)
File "C:/msys64/mingw64/lib/python3.9/site-packages/pandas/core/series.py", line 1051, in _get_value
loc = self.index.get_loc(label)
File "C:/msys64/mingw64/lib/python3.9/site-packages/pandas/core/indexes/base.py", line 3363, in get_loc
raise KeyError(key) from err
KeyError: 0
那么,如何根據對同一 DataFrame 的(復制的)切片進行的計算來更新此 DataFrame?
uj5u.com熱心網友回復:
您可以應用自定義函式將通道組減去組的第一個值。
df['value_rel'] = df.groupby('channel')['value'].apply(lambda x: x - x.iloc[0])
print(df)
# Output:
channel value value_rel
0 A 10 0
1 A 11 1
2 A 12 2
3 A 13 3
4 B 20 0
5 B 22 2
6 B 24 4
7 B 26 6
8 B 28 8
9 C 100 0
10 C 105 5
11 C 110 10
12 C 115 15
13 C 120 20
14 C 125 25
15 C 130 30
uj5u.com熱心網友回復:
您需要使用iloc,因為索引號 0 并不存在于所有tchan.
for tchan in mychans:
this_ch_data = df[df["channel"]==tchan]
df.loc[this_ch_data.index, "value_rel"] = \
this_ch_data["value"] - this_ch_data["value"].iloc[0]
也就是說,這是groupby.transformwith 的好用例first。所以不需要回圈,你可以做
df['value_rel'] = df['value'] - df.groupby('channel')['value'].transform('first')
轉載請註明出處,本文鏈接:https://www.uj5u.com/net/360936.html
