使用matplotlib做一個散點圖

Question

當我嘗試使用matplotlib來繪製代碼時，我看到一個沒有任何繪圖的空圖。我附上了代碼和空白圖。請讓我知道我缺少的東西。謝謝！

empty window with no plot

from datetime import datetime 
start_time = datetime.now() 
print(start_time) 

import pandas as pd 
import numpy as np 

file1 = 'fn_data.csv' 
import matplotlib.pyplot as plt 
#import pylab 

# Read the .txt file into a dataframe 
data = pd.read_csv(file1, encoding = "ISO-8859-1", header=0, delimiter=',') 
rating=data.iloc[:,0] 
chef=data.iloc[:,3] 
print(rating) 

mydict={} 
i = 0 
for item in chef: 
    if(i>0 and item in mydict): 
     continue 
    else: 
     i = i+1 
     mydict[item] = i 

chef_codes=[] 
for item in chef: 
    chef_codes.append(mydict[item]) 

print(chef_codes) 
chef_codes_new=np.array(chef_codes) 
rating_new=np.array(rating) 
print(type(chef_codes_new),type(rating_new)) 
print(np.max(chef_codes_new),np.max(rating_new)) 
plt.plot(kind='scatter',x=chef_codes_new,y=rating_new, marker='o', ms = 10, alpha=1, color='b') 
plt.axis([0, 1000, 0, 5]) 
plt.show() 
plt.savefig("fig1.png") 


end_time = datetime.now() 
print(end_time) 

Answer 1

def scatterPlot(X,Y): 
    ids= ['green' if y == 0 else 'red' for y in Y] 
    plt.scatter(X[:,0], X[:,1], color=ids) 
    plt.title("Scatter Plot") 
    return 

scatterPlot(x,y) 

其唯一的scatter plot一個例子希望它可以幫助你從你的問題了。

Answer 2

如果plt.scatter有效，plt.plot也可以。問題是你已經混合了pandas.DataFrame.plot的語法和pyplot.plot的命令。

所以，與其

plt.plot(kind='scatter',x=chef_codes_new,y=rating_new, marker='o', ms = 10, alpha=1, color='b') 

你需要

plt.plot(chef_codes_new,rating_new, marker='o', ms = 10, alpha=1, color='b') 

語法是plt.plot(x,y, *args, **kwargs)，你不能在一個pyplot陰謀使用kind="scatter"。如果您想要散點圖，請使用plt.scatter。