大家好。 我想在 python 中为附加的数据框编写一个函数。 我不知道我该怎么做

Question

enter image description here Hi everyone. I hope I can explain what I am going to do. I want to write a function in python for the attached data frame. the input will be different T(different names). The p-value for the input T must be less than 0.01 while for the rest of Ts must be more than 0.01, if its true the function must keep that row and then apply this on other rows too. At the end should return a df which includs col1, col2 , col3. So all this new rows have a p-value less than 0.01 for the specific T while in other Ts are more than 0.01. I appreciate any advice.

Answer 1

dropped_df = pd.DataFrame()

def check_p_value(df): #pass your dataframe
    dropped_df = df.loc[((df<0.1)).any(1)]

check_p_value(df)

df<0.1 will return a boolean df

.any(1) returns any true element over the axis 1 (rows)

df.loc[] then filters the dataframe based on the boolean df