how to subset columns if they contain a string list comprehension in python

main.py
# Import the pandas library
import pandas as pd

# Create a sample dataframe
df = pd.DataFrame({
    'fruit': ['apple', 'banana', 'orange'],
    'color': ['red', 'yellow', 'orange'],
    'shape': ['round', 'curved', 'round']
})

# Use a list comprehension to filter columns containing the string 'ru'
new_df = df[[col for col in df.columns if 'ru' in col]]

# Print the new dataframe
print(new_df)
398 chars
16 lines

main.py
     fruit   color   shape
0    apple     red   round
1   banana  yellow  curved
2   orange  orange   round
108 chars
5 lines