DataFrame возвращает фрагменты фрейма данных, значение столбца которых равно некоторому значению, иначе 0, на основе сто ⇐ Python

[Anonymous]

У меня есть кадр данных, как показано ниже

Код: Выделить всё

df = pd.DataFrame({'testid':(1,2,1,2,1,2),'Name':('apple','apple','melon','melon','orange','orange'), 'A': (1,2,10,20,5,5), 'B': (1,5,4,2,3,1)})

тест
Имя
A
B




1
яблоко
1
1

2
яблоко
2
5


1
дыня
10
4

2
дыня
20
2


1
оранжевый
5
3

2
оранжевый
5
1




Я хочу вернуть фрагмент этого фрейма данных (все еще фрейм данных) для каждого testid и столбца A и B: если соответствующее значение яблока больше 1, то оно возвращает соответствующее значение дыни, иначе возвращает 0.по сути, я хочу получить такой DataFrame




тест
A
B



1
0
< td>0


2
20
2




как этого добиться? Я попробовал применить() с лямбда-х:, но не нашел способа поместить столбец dataframe в лямбда-функцию.

Подробнее здесь: https://stackoverflow.com/questions/729 ... lue-else-0