RozwN2.py

import matplotlib.pyplot as plt
import pandas as pd

# ładowanie danych do ramki danych
data = pd.read_csv("marketing.csv")
# sortowanie
data2 = data.sort_values(by=['Facebook'])
# dodanie nowych wierszy
dataNew = pd.DataFrame({"Sprzedaz": [34, 22, 11],
                        "Facebook": [52, 42, 33],
                        "Telewizja": [33, 17, 44],
                        })
data3 = pd.concat([data2, dataNew])
# ponowne sortowanie do wykresu
data4 = data3.sort_values(by=['Facebook'])
# pierwsza czesc wykresu
plt.subplot(2, 1, 1)
plt.plot(data4["Facebook"], data4["Sprzedaz"])
plt.title("Wykres sprzedaż/facebook")
plt.ylabel("Sprzedaż")
plt.xlabel("Facebook")
# sortowanie po sprzedazy
data5 = data4.sort_values(by=['Sprzedaz'])
# druga część wykresu
plt.subplot(2, 1, 2)
plt.title("Wykres facebook i telewizja/sprzedaz")
plt.scatter(data5["Sprzedaz"], data5["Telewizja"], label="Telewizja")
plt.plot(data5["Sprzedaz"], data5["Facebook"], color="red", label="Facebook")
plt.legend()
plt.xlabel("Sprzedaż")
plt.show()