numpy.extract et numpy.any fonctions, est-il possible de le rendre plus simple?

Question

S'il y a une possibilité de simplifier ce code, je l'apprécierais vraiment! J'essaie de me débarrasser des lignes avec des zéros. La première colonne est la date. Si toutes les autres colonnes sont égales à zéro, elles doivent être supprimées. Le nombre de colonnes varie.

import numpy as np 

condition = [ np.any(list(x)[1:]) for x in r] 
r = np.extract(condition, r) 

numpy.extract docs

Answer 1

Vous pouvez éviter la compréhension de la liste et utiliser à la place FancyIndexing:

#!/usr/bin/env python 
import numpy as np 
import datetime 
r=np.array([(datetime.date(2000,1,1),0,1), 
      (datetime.date(2000,1,1),1,1), 
      (datetime.date(2000,1,1),1,0), 
      (datetime.date(2000,1,1),0,0),       
      ]) 
r=r[r[:,1:].any(axis=1)] 
print(r) 
# [[2000-01-01 0 1] 
# [2000-01-01 1 1] 
# [2000-01-01 1 0] 

si r est un ndarray, puis r [:, 1:] est une vue avec la première colonne enlevée. r [:, 1:]. Any (axis = 1) est un tableau booléen, que vous pouvez ensuite utiliser comme un "index fantaisie"