Проблемы с выбором индексации

#python #pandas #dataframe

#python #панды #фрейм данных

Вопрос:

 data = DataFrame(np.arange(16).reshape((4, 4)),
    index=['Ohio', 'Colorado', 'Utah', 'New York'],
    columns=['one', 'two', 'three', 'four'])
  

выходные данные:

 >>> data
           one  two  three  four
Ohio        0    1      2     3
Colorado    4    5      6     7
Utah        8    9     10    11
New York   12   13     14    15
  

Теперь я хочу:

  1. фильтруйте этот фрейм данных и выводите только строки, где [‘three’]> 5
  2. выводите только первые 3 строки… поэтому игнорируйте последний [: 2]

Я пишу это:

 data.iloc[data['three']>5, [:2]]
  

и получите эту ошибку:

     >>> data.iloc[data['three']>5, [:2]]
         File "<stdin>", line 1
         data.iloc[data['three']>5, [:2]]
                                     ^
  

как мне это исправить??

это мой импорт

 import pandas as pd
from pandas import Series, DataFrame
import numpy as np
  

Комментарии:

1. Попробуйте это: data[data['three']>5][:2]

2. Только что увидел, что @IoaTzimas ответил на это

Ответ №1:

Ваш вопрос немного неясен, но я понял, что вам нужны Колорадо и Юта в вашем окончательном выводе. Если это так, измените код на :

 data[data['three']>5][:2]
  

Комментарии:

1. спасибо, это слишком долго меня напрягало!