pandas logical index on columns