For loop question: Label each store for correlation value

Question

I define my code to find correlation between 'day' and 'sales' each store by using this code

def store_corr(store_num):
  subset= df[df['store']== store_num]
  subset= subset[['day','dollar_sales']]
  subset_corr = subset.corr()
  corr_val = subset_corr.iloc[0,1]
  return corr_val

And Next step is running for loop each store like this

store = df.store.unique()    
   for i in store:
      corre = store_corr(i)
      print(corre)

But All I need are columns that provide store_num and corr for each store, so How I can coding to get output like this:

store	corre
1	0.26
2	0.12
3	-0.96

Thank you

Please add some sample data, see [How to make good reproducible pandas examples](https://stackoverflow.com/q/20109391/15873043). — fsimonjetz, Sep 05 '22 at 08:04
sample data provide in this link: https://prnt.sc/RB9tx8vXPMBm — Kendo Nextbest, Sep 05 '22 at 08:08

score 1 · Answer 1 · answered Sep 05 '22 at 08:10

1

I think you could do this:

store = df.store.unique() 
values = {"store": [], "corre": []}   
for i in store:
    corre = store_corr(i)
    values["store"].append(i)
    values["corre"].append(corre)
    print(i, corre)
df = pd.DataFrame(values)

df dataframe should have the output you want

answered Sep 05 '22 at 08:10

Lucas M. Uriarte

2,403
5
19

1

If it is ok you can accept answer by clicking next to it so it get an accept answer status – Lucas M. Uriarte Sep 05 '22 at 08:16
Is that possible to pick up only the last values for this dict? because lastet list ondict has complete with all corre with store. – Kendo Nextbest Sep 05 '22 at 08:21
what do you mean by last values? – Lucas M. Uriarte Sep 05 '22 at 08:23
I mean like this sir https://prnt.sc/oam0TwP3_Imm last compute, it's have all store_num and corre – Kendo Nextbest Sep 05 '22 at 08:25
1

you have modified the print I made, change the print value by: ```print("store: ", i, "corre: ", corre)``` instead of ```print(values)``` the lastest will print the dictionary which is increasing every iteration. The initial print I made is only printing the latest – Lucas M. Uriarte Sep 05 '22 at 08:28
1

Ok sir, I got this. Thank you very much – Kendo Nextbest Sep 05 '22 at 08:28
I will appreciate if you accept my answer as valid. – Lucas M. Uriarte Sep 05 '22 at 08:30

For loop question: Label each store for correlation value

1 Answers1