我是數(shù)據(jù)科學(xué)新手，需要幫助執(zhí)行以下操作：region(I) 在我的例子中，根據(jù)列中的唯一組和另一個組分割數(shù)據(jù)集country(II) 我想將每個數(shù)據(jù)幀保存為 .csv 文件 - 像這樣regionname_country.csv，例如west_GER.csv，east_POL.csv(III) 如果可能的話，我想迭代每個 .csv 文件以繪制每個 dffor loop 的散點(diǎn)圖。education vs age(IV) 最后將我的繪圖/圖形保存在 pdf 文件中（每頁 4 個圖形）'df' Region, country, Age, Education, Income, FICO, Target1 west, GER, 43, 1, 47510, 710, 12 east, POL, 32, 2, 73640, 723, 13 east, POL, 22, 2, 88525, 610, 04 west, GER, 55, 0, 31008, 592, 05 north, USA, 19, 0, 18007, 599, 16 south, PER, 27, 2, 68850, 690, 07 south, BRZ, 56, 3, 71065, 592, 08 north, USA, 39, 1, 98004, 729, 19 east, JPN, 36, 2, 51361, 692, 010 west, ESP, 59, 1, 98643, 729, 1期望的結(jié)果： # df_to_csv : 'west_GER.csv'west, GER, 43, 1, 47510, 710, 1 west, GER, 55, 0, 31008, 592, 0 # west_ESP.csvwest, ESP, 59, 1, 98643, 729, 1 # east_POL.csveast, POL, 32, 2, 73640, 723, 1 ...# north_USA.csvnorth, USA, 39, 1, 98004, 729, 1 north, USA, 19, 0, 18007, 599, 1
2 回答
呼如林
TA貢獻(xiàn)1798條經(jīng)驗 獲得超3個贊
對于 Python：（
一）和（二）：
for i in df.groupby(["Region", "country"])[["Region", "country"]].apply(lambda x: list(np.unique(x))):
df.groupby(["Region", "country"]).get_group((i[1], i[0])).to_csv(f"{i[1]}_{i[0]}.csv")
（三）、（四）：
import glob
import matplotlib.pyplot as plt
fig, axs = plt.subplots(nrows=2, ncols=2)
for ax, file in zip(axs.flatten(), glob.glob("./*csv")):
df_temp = pd.read_csv(file)
region_temp = df_temp['Region'][0]
country_temp = df_temp['country'][0]
ax.scatter(df_temp["Age"], df_temp["Education"])
ax.set_title(f"Region:{region_temp}, Country:{country_temp}")
ax.set_xlabel("Age")
ax.set_ylabel("Education")
plt.tight_layout()
fig.savefig("scatter.pdf")
慕俠2389804
TA貢獻(xiàn)1719條經(jīng)驗 獲得超6個贊
在 R 中，您可以這樣做：
library(tidyverse)
#get data in list of dataframes
df %>%
select(Region, country, Education, Age) %>%
group_split(Region, country) -> split_data
#From list of data create list of plots.
list_plots <- map(split_data, ~ggplot(.) + aes(Education, Age) +
geom_point() +
ggtitle(sprintf('Plot for region %s and country %s',
first(.$Region), first(.$country))))
#Write the plots in pdf as well as write the csvs.
pdf("plots.pdf", onefile = TRUE)
for (i in seq_along(list_plots)) {
write.csv(split_data, sprintf('%s_%s.csv',
split_data[[i]]$Region[1], split_data[[i]]$country[1]), row.names = FALSE)
print(list_plots[[i]])
}
dev.off()
添加回答
舉報
0/150
提交
取消