I have a pandas DataFrame with 3 columns, shown below.
col1 value flag 1 0 0 2 0.03915 0 3 0.13 1
I want to create a scatterplot from this dataframe where col1
is the x axis and value
is the y axis, and the part I'm struggling to do is, for the rows that have flag=0
I want the color of this point to be blue and similarly if flag=1
I want to color the point red.
Is there a simple to to check the flag column per row and color the point accordingly?
Assuming the dataframe containing the given data is df
, this is what you want. You can create a list of colors according to your condition flag
column and its values. Feed that colors list to color
argument in the built-in DataFrame.plot.scatter
function. (You can find the docs here .)
colors = ['r' if flag==1 else 'b' for flag in df.flag]
df.plot.scatter('col1', 'value', color=colors)
Hope this helps.
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.