lasse/elplot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Made it easier to set colors.

Browse Source
This commit is contained in:
Lasse Edfast 2023-01-24 10:59:09 +01:00 committed by GitHub
parent 61407e7d58
commit 92378ff191
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
dotplot.py
View File

@ -11,8 +11,10 @@ df['hour']=df.index
df['y']=df['hour'].apply(lambda x: int(x/120)) df['y']=df['hour'].apply(lambda x: int(x/120))
df['x']=df['hour'].apply(lambda x: int(x%120)) df['x']=df['hour'].apply(lambda x: int(x%120))
# Set export to orange and import to grey. # Set colors for export and import
df['color'] = df['balance'].apply(lambda x: 'orange' if x < 0 else 'grey') export_color = 'orange'
import_color = 'grey'
df['color'] = df['balance'].apply(lambda x: export_color if x < 0 else import_color)
# Plot the data. # Plot the data.
ax = df.plot(kind='scatter', x='x', y='y', c='color') ax = df.plot(kind='scatter', x='x', y='y', c='color')