pandas - 如何使用熊猫在excel中随机放置数字、字符串

Question

我在excel中有3000行数据

id,product,store,revenue,data,state
1,Ball,,222,nil,,
1,Pen,,234,nil,nil,,
2,Books,,543,nil,,
2,Ink,,123,nil,,

• 我需要store用随机数填充第三列1 & 5

我的代码每次都给出 1，df['store'] = df['store'].fillna(random.randint(1,5))

• 我需要state用随机字符串填充第 5 列{'CA', 'WD','CH', 'AL'}

country如果第 5 列中的“CA”、“CH”映射到美国，“WD”、“AL”映射到日本， 我需要创建第 6列{'CA':'USA', 'CH':'USA', 'AL':'Japan'}

标签： pandas

Answer 1

让我们试试

n=len(df)
num=np.random.randint(1,6,size=n)
l={'CA', 'WD','CH', 'AL'}
state=np.random.choice(list(l), n)

df['store'] = df['store'].fillna(pd.Series(num,index=df.index))
df['state'] = df['state'].fillna(pd.Series(state,index=df.index))
df['country']=df.state.map({'CA':'USA', 'CH':'USA', 'AL':'Japan'})