Skip to content
Advertisement

Is there a way to mimic the pipe operator from R in Python?

See the example below. Is it possible to put each Python operation on a new line? If I have many operations, the code is not all visible. I need to scroll the VS Code window to the right. Thanks

# R code
data %>%
  group_by(sex) %>%
  summarise(avg_height=mean(height))

# Python code
data.groupby(['sex']).agg(avg_height=('height', 'mean')).reset_index()

Advertisement

Answer

In Python, line breaks inside parentheses are ignored, so you could rewrite as:

(data
    .groupby(['sex'])
    .agg(avg_height=('height', 'mean'))
    .reset_index())

This post may be helpful.

User contributions licensed under: CC BY-SA
9 People found this is helpful
Advertisement