Python - Obtener datos de arrays

Hola, tengo una duda de como seleccionar elementos en un array

pongo un ejemplo


tengo unos datos que son asi:

fechas unidades1 unidades2 unidades3 unidades4 unidades5
0 fecha1 verde azul azul azul rojo
1 fecha2 verde verde verde verde azul
2 fecha3 amarillo amarillo azul amarillo verde
3 fecha4 rojo verde azul rojo azul
4 fecha5 amarillo amarillo verde rojo verde
5 fecha6 verde verde azul verde azul
6 fecha7 amarillo amarillo azul azul azul

quiero obtener todas las fechas donde en unidades ponga rojo
osea
fecha1, fecha 4, fecha5
pero sin utlizar bucles


si hago
me saldria

array(['fecha4'], dtype=object)

que es el resultado buscado si miro solo las unidades1

quiero hacer lo mismo con todas las columnas

y como no puedo utilizar bucles:

etc

pero como lo haria para hacerlo de golpe sin poner 0,1,2,3.....porque eso es poner un bucle a mano y si hay un millon.....

Hola Marta, los datos como están? en una lista de listas?? nos lo puedes poner para poder hacer pruebas?

Puedes utilizar la función filter? https://docs.python.org/3.5/library/functions.html#filter

Los datos vienen en un csv.

Para realizar el ejercicio solo puedo usar numpy
(y panda para cargar el csv)

y no se pueden usar bucles.

Lo mismo tengo q meter el [:,0][:,1][:,2]....etc.... a mano, pero me parece raro.