Как загрузить данные для барплота из кадра данных с помощью matplotlib - PullRequest
Новая
       12

Как загрузить данные для барплота из кадра данных с помощью matplotlib

1 голос
/ 07 мая 2019

Итак, у меня есть фрейм данных с именем df1, который я использовал pandas для импорта файла CSV в свой блокнот Jupyter.Теперь я хочу отобразить результаты этого информационного кадра в виде гистограммы.На данный момент я просто делаю это вручную, вставляя цифры следующим образом: data = [120,52,50].Как вместо этого ссылаться на фрейм данных df1?Я пытался, но это не так просто, как должно быть.

Это фрейм данных под названием df1: 

+---+---------+---------+
|   | column1 | column2 |
+---+---------+---------+
| 0 |      A  |   120   |
| 1 |      B  |   52    |
| 2 |      C  |   50    |
+---+---------+---------+

    import matplotlib, matplotlib.pyplot as plt, numpy as np, random as rd, 
    pandas as pd, seaborn as sns
    %matplotlib inline

# import CSV to dataframe

    df1 = pd.read_csv("test1.csv")
    df1

# generate barplot

    data = [120,52,50]
    columns = 0,1,2
    bars = ("bar1", "bar2", "bar3")
    plt.figure(figsize = (8,6))
    b = plt.bar(columns,data,width = 0.8, color = ('#569cff','#82b5ff','#b5d1ff'))
    plt.xticks(columns,bars)
    plt.legend(b,bars,fontsize=14)
    plt.title("title goes here",fontsize=16)
    plt.ylabel('y label',fontsize=12)
    plt.xlabel('x label',fontsize=12)
    plt.show()

1 Ответ

0 голосов
/ 07 мая 2019

Если вы просто ищете простую гистограмму, попробуйте: 

df=pd.DataFrame({'column1': ('A','B','C'),'column2':(120,52,50)}).set_index('column1')
df.plot(kind='bar')
...