如何创建一个具有任意字符串的新列

Question

我试图用随机字符串向我的数据文件中添加一个新列，如何使randomword的输出与withColumn的输入保持一致？

def randomword(length):
   letters = string.ascii_lowercase
   return ''.join(random.choice(letters) for i in range(length))

df_with_new_col = mydf.withColumn('random_string', randomword(10))

Answer 1

''.join(random.choices(string.ascii_lowercase, k=N))

random.choices Python 3.6+