初心者 python DataFrame 教えてください。

Question

import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.randn(1000,4),columns=["one","two","three","four"])

　5列目に英字（大文字と小文字）をランダムで表示したいのですが方法がわかりません。
列の最初（fourの横はkey）でお願いします。
ご教授お願いします。

python
コード
ran = random.choice(string.ascii_letters)
for i in range(1000):
　　　　a = random.choice(string.ascii_letters)
　　　　df ["key"] = a

続きはこんな感じでやってみましたがうまくできませんでした。
5列目を追加した後にcsvファイルに書き込むところまでやるのが今回の最終目的です。

Answer 1

列の追加の右辺はリストやタプルや一次元のnumpy.ndarrayなどですので、以下のようにやってください。

>>> import pandas as pd
>>> import numpy as np
>>> import string
>>> import random
>>>
>>> df = pd.DataFrame(np.random.randn(1000,4),columns=["one","two","three","four"])
>>> df["key"] = [random.choice(string.ascii_letters) for i in range(1000)]
>>> print(df)
          one       two     three      four key
0    0.029644  0.100708 -1.095874 -0.825876   D
1    0.753252 -1.164001  1.397749  0.206278   l
2   -0.703004  1.658549 -0.525967  0.963872   G
3    0.853257 -1.376528  0.392412 -1.638464   d
4   -0.556479 -1.052158 -0.007249  0.758621   g
..        ...       ...       ...       ...  ..
995  0.338425  0.567574  1.038987  1.044566   Z
996 -0.310239  0.484861  0.714648 -0.876791   U
997 -1.103171 -0.410534 -0.930673 -1.229390   t
998 -1.048326 -1.142197  0.138967  0.537495   z
999 -0.192669 -0.721866 -0.120771  0.506251   V

[1000 rows x 5 columns]

Comments

[kuriyamasaikou]

ありがとうございました。

Answer 2

Python3.6以降の場合

import pandas as pd
import numpy as np
import random
import string

df = pd.DataFrame(np.random.randn(1000,4),columns=["one","two","three","four"])
df['key'] = random.choices(string.ascii_letters, k=1000)
print(df.head())
#	one	two	three	four	key
#0	-1.863726	-1.598030	0.590290	1.000697	a
#1	-0.963442	-0.651434	-0.444132	-0.017479	T
#2	-0.991826	1.166598	2.065291	-0.825682	b
#3	0.789025	-0.978161	0.170264	0.456968	W
#4	0.369024	-0.528037	-0.674172	-1.290081	D

Comments

[kuriyamasaikou]

ありがとうございました。解決できました！