Блог

Я хотел бы напечатать (или вернуть) только 3 верхние строки фрейма данных pandas. Выполнение следующего дает мне все из них:

 df1 = pd.DataFrame(np.random.randn(10,2),columns=list('AB'))
df2 = df1.sort(["B"], ascending=[True]) 

df2.to_json(orient="records")  # prints all the rows
  

Как мне ограничить ТОЛЬКО 3 верхние строки и распечатать их в формате JSON?

Спасибо!

Вы должны иметь возможность использовать head и передавать количество строк для включения

 short_list = df2.head(3).to_json(orient="records")
  

Вывод:

 '[
  {"A":-0.4056731092,"B":-1.2808121187},
  {"A":-0.1866904667,"B":-1.2226082762},
  {"A":0.462008584,"B":-1.070959354}
 ]'