Tengo un df:
col1 col2 col3
0 1 2 3
1 2 3 1
2 3 3 3
3 4 3 2
Quiero agregar una nueva columna basada en las siguientes condiciones:
- if col1 > col2 > col3 -----> 2
- elif col1 > col2 -----> 1
- elif col1 < col2 < col3 -----> -2
- elif col1 < col2 -----> -1
- else -----> 0
Y debería convertirse en esto:
col1 col2 col3 new
0 1 2 3 -2
1 2 3 1 -1
2 3 3 3 0
3 4 3 2 2
Seguí el método de esta publicación por unutbu , con 1 mayor o menor que está bien. Pero en mi caso con más de 1 mayor o menor que , las condiciones devuelven un error:
conditions = [
(df['col1'] > df['col2'] > df['col3']),
(df['col1'] > df['col2']),
(df['col1'] < df['col2'] < df['col3']),
(df['col1'] < df['col2'])]
choices = [2,1,-2,-1]
df['new'] = np.select(conditions, choices, default=0)
Traceback (most recent call last):
File "<ipython-input-43-768a4c0ecf9f>", line 2, in <module>
(df['col1'] > df['col2'] > df['col3']),
File "C:\ProgramData\Anaconda3\lib\site-packages\pandas\core\generic.py", line 1478, in __nonzero__
.format(self.__class__.__name__))
ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().
¿Cómo debería hacer esto?
Cambia tu código a
conditions = [
(df['col1'] > df['col2']) & (df['col2'] > df['col3']),
(df['col1'] > df['col2']),
(df['col1'] < df['col2']) & (df['col2'] < df['col3']),
(df['col1'] < df['col2'])]
choices = [2,1,-2,-1]
df['new'] = np.select(conditions, choices, default=0)
Este artículo se recopila de Internet, indique la fuente cuando se vuelva a imprimir.
En caso de infracción, por favor [email protected] Eliminar
Déjame decir algunas palabras