Potresti usare np.where . Se condè un array booleano e Ae Bsono array, allora
C = np.where(cond, A, B)
definisce C come uguale a Adove condè True e Bdove condè False.
import numpy as np
import pandas as pd
a = [['10', '1.2', '4.2'], ['15', '70', '0.03'], ['8', '5', '0']]
df = pd.DataFrame(a, columns=['one', 'two', 'three'])
df['que'] = np.where((df['one'] >= df['two']) & (df['one'] <= df['three'])
, df['one'], np.nan)
rendimenti
one two three que
0 10 1.2 4.2 10
1 15 70 0.03 NaN
2 8 5 0 NaN
Se hai più di una condizione, puoi invece usare np.select . Ad esempio, se desideri df['que']eguagliare df['two']quando df['one'] < df['two'], allora
conditions = [
(df['one'] >= df['two']) & (df['one'] <= df['three']),
df['one'] < df['two']]
choices = [df['one'], df['two']]
df['que'] = np.select(conditions, choices, default=np.nan)
rendimenti
one two three que
0 10 1.2 4.2 10
1 15 70 0.03 70
2 8 5 0 NaN
Se possiamo presumere che df['one'] >= df['two']quando df['one'] < df['two']è False, le condizioni e le scelte potrebbero essere semplificate
conditions = [
df['one'] < df['two'],
df['one'] <= df['three']]
choices = [df['two'], df['one']]
(L'assunzione potrebbe non essere vera se df['one']o df['two']contengono NaN.)
Nota che
a = [['10', '1.2', '4.2'], ['15', '70', '0.03'], ['8', '5', '0']]
df = pd.DataFrame(a, columns=['one', 'two', 'three'])
definisce un DataFrame con valori stringa. Dal momento che sembrano numerici, potrebbe essere meglio convertire quelle stringhe in float:
df2 = df.astype(float)
Ciò cambia i risultati, tuttavia, poiché le stringhe confrontano carattere per carattere, mentre i float vengono confrontati numericamente.
In [61]: '10' <= '4.2'
Out[61]: True
In [62]: 10 <= 4.2
Out[62]: False
ifaffermazione èFalse?