plot rectangular wave python

Question

I have a dataframe that looks like this:

df:

    Start     End  Value
0       0   98999      0
1   99000  101999      1
2  102000  155999      0
3  156000  161999      1
4  162000  179999      0

I would like to plot a rectangular wave that goes from "Start" to "End" and that has the value in "Value" Could you please help me? thanks!

My attempt:

    for j in range(0,df.shape[0]):
        plt.figure()
        plt.plot(range(df['Start'].iloc[j],df['End'].iloc[j]), np.ones(df['End'].iloc[j]-df['Start'].iloc[j])*df['Value'].iloc[j])

but it plots in different figures...

It's ploting in different figure because you have the plt.figure() in the loop — Tbaki
– Tbaki, Commented Jun 23, 2017 at 14:55

Nick is tired · Accepted Answer · 2017-06-23 15:33:14Z

1

Maybe you can use plt.step to plot a stepwise function:

import pandas as pd
from matplotlib import pyplot as plt

df = pd.DataFrame({'End': [98999, 101999, 155999, 161999, 179999],
 'Start': [0, 99000, 102000, 156000, 162000],
 'Value': [0, 1, 0, 1, 0]})

plt.step(df.Start, df.Value, where='post')

This will plot every Value from its corresponding Start value to the next Start.

edited Jun 23, 2017 at 15:33

Nick is tired

7,17721 gold badges44 silver badges55 bronze badges

answered Jun 23, 2017 at 14:51

gereleth

2,51215 silver badges23 bronze badges

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

Comments

Tbaki · Accepted Answer · 2017-06-23 14:54:05Z

0

A bit slow...

import matplotlib.pyplot as plt 
l = df.apply(lambda x: ( [x.Value for i in range(x.Start,x.End)] ),axis=1).sum()
plt.plot(l)
plt.show()

edited Jun 23, 2017 at 14:54

answered Jun 23, 2017 at 14:38

Tbaki

1,0037 silver badges12 bronze badges

Comments

Cedric Zoppolo · Accepted Answer · 2017-06-23 15:01:13Z

0

I believe step function under matplotlib.pyplot is what you are looking for.

You can try out this code:

import pandas as pd
import matplotlib.pyplot as plt
df = pd.DataFrame([[0, 98999, 0],[99000, 101999, 1],[102000, 155999, 0],[156000, 161999, 1],[162000, 179999, 0]],columns=["Start","End","Value"])
plt.step(df["Start"],df["Value"],where="post")
plt.show()

This will end up showing below graph:

answered Jun 23, 2017 at 15:01

Cedric Zoppolo

4,7937 gold badges34 silver badges63 bronze badges

Collectives™ on Stack Overflow

plot rectangular wave python

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related