Как создать точечную диаграмму столбца из двух разных фреймов данных? - PullRequest
Новая
       25

Как создать точечную диаграмму столбца из двух разных фреймов данных?

0 голосов
/ 28 апреля 2019

У меня есть фрейм данных 1377x6 с двумя интересующими столбцами: «ID» и «foo» Я разделил столбец foo на> 0,25 и <0,25, в результате чего получилось 2 разных кадра данных. </p>

Я хочу создать диаграмму разброса значений из> 0,25 кадра данных и <0,25 кадра данных. </p>

Я решил, что у меня не возникнет много проблем, если я просто объединю отрицательный столбец с положительным фреймом данных, но мне было интересно можно ли создать график рассеяния столбца, используя значения из двух разных фреймов данных?

Я пытался использовать plt.scatter(x = greater['foo'], y = less['foo']) но я получил ValueError, говоря, что x и y должны быть одинакового размера.

Решит ли проблема объединение столбцов в один фрейм данных?

1 Ответ

0 голосов
/ 29 апреля 2019

Если вы можете использовать seaborn, я думаю, что вы хотите sns.swarmplot: 

import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
import numpy as np

# generate dataframe with 100 random integers between 0 and 100
df = pd.DataFrame({'a': np.random.randint(0,100,100)})

# create a new column with True if value is larger than 50 and False otherwise
df['larger_than_50'] = df['a'] > 50

# make swarmplot
sns.swarmplot(x='larger_than_50', y='a', data=df)
plt.show()
...