[英]Adding rows to DataFrame conditioned on values
我有以下Pandas DataFrame:
start_timestamp_milli end_timestamp_milli name rating
1 1555414708025 1555414723279 Valence 2
2 1555414708025 1555414723279 Arousal 6
3 1555414708025 1555414723279 Dominance 2
4 1555414708025 1555414723279 Sadness 1
5 1555414708025 1555414723279 Happiness 0
6 1555414708025 1555414723279 Anger 0
7 1555414708025 1555414723279 Surprise 0
8 1555414708025 1555414723279 Stress 0
9 1555414813304 1555414831795 Valence 3
10 1555414813304 1555414831795 Arousal 5
11 1555414813304 1555414831795 Dominance 2
12 1555414813304 1555414831795 Sadness 0
13 1555414813304 1555414831795 Happiness 0
14 1555414813304 1555414831795 Anger 0
15 1555414708025 1555414723279 Surprise 0
16 1555414708025 1555414723279 Stress 0
17 1555414921819 1555414931382 Valence 1
18 1555414921819 1555414931382 Arousal 7
19 1555414921819 1555414931382 Dominance 2
20 1555414921819 1555414931382 Sadness 1
21 1555414921819 1555414931382 Happiness 0
22 1555414921819 1555414931382 Anger 1
23 1555414708025 1555414723279 Surprise 0
24 1555414708025 1555414723279 Stress 1
現在,對於具有相同start_timestamp_milli
和end_timestamp_milli
每個塊,我想插入一個名為“Neutral”的額外行,如果悲傷,幸福,憤怒,驚喜和壓力的等級為0,則為等級1,否則為0。新行的start_timestamp_milli
和end_timestamp_milli
應設置為該塊的值。
生成的DataFrame應如下所示:
start_timestamp_milli end_timestamp_milli name rating
1 1555414708025 1555414723279 Valence 2
2 1555414708025 1555414723279 Arousal 6
3 1555414708025 1555414723279 Dominance 2
4 1555414708025 1555414723279 Sadness 1
5 1555414708025 1555414723279 Happiness 0
6 1555414708025 1555414723279 Anger 0
7 1555414708025 1555414723279 Surprise 0
8 1555414708025 1555414723279 Stress 0
9 1555414708025 1555414723279 Neutral 0
10 1555414813304 1555414831795 Valence 3
11 1555414813304 1555414831795 Arousal 5
12 1555414813304 1555414831795 Dominance 2
13 1555414813304 1555414831795 Sadness 0
14 1555414813304 1555414831795 Happiness 0
15 1555414813304 1555414831795 Anger 0
16 1555414708025 1555414723279 Surprise 0
17 1555414708025 1555414723279 Stress 0
18 1555414708025 1555414723279 Neutral 1
19 1555414921819 1555414931382 Valence 1
20 1555414921819 1555414931382 Arousal 7
21 1555414921819 1555414931382 Dominance 2
22 1555414921819 1555414931382 Sadness 1
23 1555414921819 1555414931382 Happiness 0
24 1555414921819 1555414931382 Anger 1
25 1555414708025 1555414723279 Surprise 0
26 1555414708025 1555414723279 Stress 1
27 1555414708025 1555414723279 Neutral 0
如何才能做到這一點?
您可以使用過濾器之前做groupby
agg
+ all
,那么concat
回結果
s=df.loc[df.name.isin(['Sadness', 'Happiness', 'Anger', 'Surprise' , 'Stress']),'rating'].\
eq(0).\
groupby([df['start_timestamp_milli'],df['end_timestamp_milli']]).\
agg('all').reset_index().assign(name='Neutral')
df=pd.concat([df,s],sort=False).sort_values(['start_timestamp_milli','end_timestamp_milli'])
df
Out[66]:
start_timestamp_milli end_timestamp_milli name rating
1 1555414708025 1555414723279 Valence 2
2 1555414708025 1555414723279 Arousal 6
3 1555414708025 1555414723279 Dominance 2
4 1555414708025 1555414723279 Sadness 1
5 1555414708025 1555414723279 Happiness 0
6 1555414708025 1555414723279 Anger 0
7 1555414708025 1555414723279 Surprise 0
8 1555414708025 1555414723279 Stress 0
15 1555414708025 1555414723279 Surprise 0
16 1555414708025 1555414723279 Stress 0
23 1555414708025 1555414723279 Surprise 0
24 1555414708025 1555414723279 Stress 1
0 1555414708025 1555414723279 Neutral 0
9 1555414813304 1555414831795 Valence 3
10 1555414813304 1555414831795 Arousal 5
11 1555414813304 1555414831795 Dominance 2
12 1555414813304 1555414831795 Sadness 0
13 1555414813304 1555414831795 Happiness 0
14 1555414813304 1555414831795 Anger 0
1 1555414813304 1555414831795 Neutral 1
17 1555414921819 1555414931382 Valence 1
18 1555414921819 1555414931382 Arousal 7
19 1555414921819 1555414931382 Dominance 2
20 1555414921819 1555414931382 Sadness 1
21 1555414921819 1555414931382 Happiness 0
22 1555414921819 1555414931382 Anger 1
2 1555414921819 1555414931382 Neutral 0
我會先pivot
,這使查找更容易,然后調整您的初始DataFrame。
f = (df.pivot_table(index=['start_timestamp_milli', 'end_timestamp_milli'],
columns='name', values='rating', aggfunc='any', fill_value=0).astype(int))
現在找到時間戳組合:
cols = ['Sadness', 'Happiness', 'Anger', 'Surprise', 'Stress']
appd = pd.Series(np.where(f[cols].any(1), 0, 1), index=f.index)
res = pd.concat([df, appd.rename('rating').reset_index().assign(name='Neutral')])
end_timestamp_milli name rating start_timestamp_milli
1 1555414723279 Valence 2 1555414708025
2 1555414723279 Arousal 6 1555414708025
3 1555414723279 Dominance 2 1555414708025
4 1555414723279 Sadness 1 1555414708025
5 1555414723279 Happiness 0 1555414708025
6 1555414723279 Anger 0 1555414708025
7 1555414723279 Surprise 0 1555414708025
8 1555414723279 Stress 0 1555414708025
9 1555414831795 Valence 3 1555414813304
10 1555414831795 Arousal 5 1555414813304
11 1555414831795 Dominance 2 1555414813304
12 1555414831795 Sadness 0 1555414813304
13 1555414831795 Happiness 0 1555414813304
14 1555414831795 Anger 0 1555414813304
15 1555414723279 Surprise 0 1555414708025
16 1555414723279 Stress 0 1555414708025
17 1555414931382 Valence 1 1555414921819
18 1555414931382 Arousal 7 1555414921819
19 1555414931382 Dominance 2 1555414921819
20 1555414931382 Sadness 1 1555414921819
21 1555414931382 Happiness 0 1555414921819
22 1555414931382 Anger 1 1555414921819
23 1555414723279 Surprise 0 1555414708025
24 1555414723279 Stress 1 1555414708025
0 1555414723279 Neutral 0 1555414708025
1 1555414831795 Neutral 1 1555414813304
2 1555414931382 Neutral 0 1555414921819
這會將它們添加到Frame的末尾,如果您希望它們間歇放置,您需要進行排序。
你可以試試:
# turn to long table with columns being feelings
new_df = (df.pivot_table(index=['start_timestamp_milli', 'end_timestamp_milli',],
columns='name',
values='rating')
.fillna(0)
)
new_df['Neutral'] = (new_df[['Sadness','Happiness','Anger', 'Surprise', 'Stress']]
.eq(0).all(axis=1).astype(int)
)
new_series = new_df[['Neutral']].stack()
new_series.name = 'rating'
df = pd.concat((df, new_series.reset_index()))
IIUC:
se = ['start_timestamp_milli', 'end_timestamp_milli']
names = 'Sadness, Happiness, Anger, Surprise, Stress'.split(', ')
創造這個小家伙
neutral = df.set_index(se).query('name in @names').rating.eq(0).all(level=[0, 1]).mul(1)
neutral
start_timestamp_milli end_timestamp_milli
1555414708025 1555414723279 0
1555414813304 1555414831795 1
1555414921819 1555414931382 0
Name: rating, dtype: int64
然后使用值'Neutral'
為其添加name
列,並將其附加到原始列。 不要忘記排序和丟棄東西,使其再次漂亮。
df.append(neutral.reset_index().assign(name='Neutral'), sort=False) \
.sort_values(se).reset_index(drop=True)
start_timestamp_milli end_timestamp_milli name rating
0 1555414708025 1555414723279 Valence 2
1 1555414708025 1555414723279 Arousal 6
2 1555414708025 1555414723279 Dominance 2
3 1555414708025 1555414723279 Sadness 1
4 1555414708025 1555414723279 Happiness 0
5 1555414708025 1555414723279 Anger 0
6 1555414708025 1555414723279 Surprise 0
7 1555414708025 1555414723279 Stress 0
8 1555414708025 1555414723279 Surprise 0
9 1555414708025 1555414723279 Stress 0
10 1555414708025 1555414723279 Surprise 0
11 1555414708025 1555414723279 Stress 1
12 1555414708025 1555414723279 Neutral 0
13 1555414813304 1555414831795 Valence 3
14 1555414813304 1555414831795 Arousal 5
15 1555414813304 1555414831795 Dominance 2
16 1555414813304 1555414831795 Sadness 0
17 1555414813304 1555414831795 Happiness 0
18 1555414813304 1555414831795 Anger 0
19 1555414813304 1555414831795 Neutral 1
20 1555414921819 1555414931382 Valence 1
21 1555414921819 1555414931382 Arousal 7
22 1555414921819 1555414931382 Dominance 2
23 1555414921819 1555414931382 Sadness 1
24 1555414921819 1555414931382 Happiness 0
25 1555414921819 1555414931382 Anger 1
26 1555414921819 1555414931382 Neutral 0
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.