嘗試遍歷一個(gè) for 循環(huán)，該循環(huán)在 pandas 數(shù)據(jù)幀上運(yùn)行 3 個(gè)回歸，同時(shí)為每個(gè)變量打印一條線圖。year = crime_df.iloc[:,0]violent_crime_rate = crime_df.iloc[:,3]murder_rate = crime_df.iloc[:,5]aggravated_assault_rate = crime_df.iloc[:,11]x_axis = [violentcrimerate, murderrate, aggravatedassaultrate]for x in x_axis: slope, intercept, r_value, p_value, std_err = linregress(year, x) fit = slope * year + intercept fig, ax = plt.subplots() fig.suptitle('x', fontsize=16, fontweight="bold") ax.plot(year, x, linewidth=0, marker='o') ax.plot(year, fit, 'b--') plt.show()代碼生成 3 個(gè)標(biāo)題為“x”的圖和不同的回歸線，但我想知道如何為每個(gè)圖設(shè)置相對(duì)于循環(huán)中每個(gè)變量的相對(duì)標(biāo)題（和標(biāo)簽）。不確定如何從我引用的列表中檢索變量名稱。在字幕行中嘗試過str(x)，但返回的是列中的值而不是列表標(biāo)題。
1 回答
鳳凰求蠱
TA貢獻(xiàn)1825條經(jīng)驗(yàn) 獲得超4個(gè)贊
像這樣的東西？
import numpy as np
import matplotlib.pyplot as plt
matrix = np.random.rand(4,12) # emulate some data
crime_df = pd.DataFrame(matrix)# emulate some data
year = crime_df.iloc[:,0]
violent_crime_rate = crime_df.iloc[:,3]
murder_rate = crime_df.iloc[:,5]
aggravated_assault_rate = crime_df.iloc[:,11]
names = ['violent_crime_rate','murder_rate','aggravated_assault_rate']
x_axis = [violent_crime_rate, murder_rate, aggravated_assault_rate]
def linregress(year,x): #emulate some data
return np.random.rand(5)
for ind, x in enumerate(x_axis):
slope, intercept, r_value, p_value, std_err = linregress(year, x)
fit = slope * year + intercept
fig, ax = plt.subplots()
fig.suptitle('x:'+str(names[ind]), fontsize=16, fontweight="bold")
ax.plot(year, x, linewidth=0, marker='o', label = names[ind] + ':1')
ax.plot(year, fit, 'b--', label = names[ind] + ':2')
ax.legend()
plt.show()
添加回答
舉報(bào)
0/150
提交
取消