ฉันมีชุดข้อมูลเช่นนี้:
ตัวอย่าง Dataframe
import pandas as pd
df = pd.DataFrame({
'names': ['A','B','C','D','E','F','G','H','I','J','K','L'],
'col1': [0, 1, 0, 1, 1, 1, 0, 0, 0, 1, 0, 0],
'col2': [0, 0, 0, 0, 1, 0, 1, 0, 1, 0, 0, 0]})
ฉันต้องการแทนที่บางส่วน0
ในcol1
และcol2
ด้วย1
's แต่ไม่แทนที่0
' s หากสามหรือมากกว่า0
อยู่ติดกันในคอลัมน์เดียวกัน วิธีนี้สามารถทำได้กับแพนด้า
ชุดข้อมูลดั้งเดิม:
names col1 col2
A 0 0
B 1 0
C 0 0
D 1 0
E 1 1
F 1 0
G 0 1
H 0 0
I 0 1
J 1 0
K 0 0
L 0 0
ที่ต้องการอยากมีชุดข้อมูล:
names col1 col2
A 1 0
B 1 0
C 1 0
D 1 0
E 1 1
F 1 1
G 0 1
H 0 1
I 0 1
J 1 0
K 1 0
L 1 0
df.loc[(df['col1']+df['col1'].shift(1)+df['col1'].shift(2)>0)&(df['col1']+df['col1'].shift(1)+df['col1'].shift(-1)>0)&(df['col1']+df['col1'].shift(-1)+df['col1'].shift(-2)>0)]=1
อย่างไรก็ตามสิ่งนี้จะทำให้แถวแรกและแถวสุดท้ายไม่มีการแตะต้อง
col2
ไงล่ะ