Iterate through columns in a dataframe and plot on the same graph (Python)

Question

I am relatively new to Python, although I have some basic understanding of data analysis and statistics from my experience as a molecular biologist. With that, I am trying to use Python to plot several columns I have in my dataframe into one nice graph to be able to compare the individual data from various experimental conditions. I have attached an example in a screenshot showing how I have some columns after I did some very basic data wrangling and cleaning (transpose, removing NaN or '0' values, etc.). As you can see, I'd like to be able to loop through columns (let's say columns 2-10 for example) and plot only that data into the kde graph. How can I go about doing this efficiently? Thank you so much for your patience and help! enter image description here

sns.set_style("darkgrid")
sns.kdeplot(data=df['N01_POOL4_1143_totNG_120_sampNG'],shade=True)
sns.kdeplot(data=df['N01_POOL9_2417_totNG_636_sampNG'],shade=True)
sns.kdeplot(data=df['N15_POOL6_1889_totNG_60_sampNG'],shade=True)
sns.kdeplot(data=df['N95_POOL2_669_totNG_75_sampNG'],shade=True)
sns.kdeplot(data=df['PosC1U_POOL2_669_totNG_171_sampNG'],shade=True)
plt.xlim(-400,3000)
plt.ylim(-0.00001,0.004)

Alexandra Dudkina · Accepted Answer · 2020-09-27 21:08:21Z

2

You can create a list slice and iterate over it:

for col_name in df.columns.values[1:10]:
  sns.kdeplot(data=df[col_name],shade=True)

Here df1.columns.values returns dataframe column names as list of strings and [1:10] takes a sublist from first to tenth element.

answered Sep 27, 2020 at 21:08

Alexandra Dudkina

4,5123 gold badges18 silver badges29 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

srajpara Over a year ago

Thank you SO much!! This was very helpful.

Collectives™ on Stack Overflow

Iterate through columns in a dataframe and plot on the same graph (Python)

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related