Python中使用多个np.where拼接字符串遇UFuncTypeError问题求助
解决pandas中np.where字符串拼接触发UFuncTypeError的问题
我需要在Python/pandas里生成供其他程序使用的数据键,基于DataFrame现有列的调整值构建。单独执行以下代码能正常运行:
df_mpf['DATAKEY'] = np.where(df_mpf['SEX']==1,'M','F').astype(str)
但拼接两个np.where结果时:
df_mpf['DATAKEY'] = np.where(df['SEX']==1,'M','F').astype(str) + np.where(df['SEX']==1,'M','F').astype(str)
触发错误:
UFuncTypeError: ufunc 'add' did not contain a loop with signature matching types (dtype('<U1'), dtype('<U1')) -> dtype('<U1')
奇怪的是,把np.where结果和TERM_Y列表或df_mpf['AGE']列拼接时却没问题:
df_mpf['DATAKEY'] = np.where(df_mpf['SEX']==1,'M','F').astype(str) + TERM_Y.astype(str) + df_mpf['AGE'].astype(str)
解决方法
方案1:将np.where结果转为pandas Series后拼接
np.where返回的是numpy数组,直接用+拼接同类型字符串数组可能触发numpy类型匹配问题,转为Series后用pandas的字符串拼接逻辑即可解决:sex_str = pd.Series(np.where(df_mpf['SEX']==1,'M','F'), dtype=str) df_mpf['DATAKEY'] = sex_str + sex_str方案2:使用numpy.char.add方法
numpy的char模块专门处理字符串数组操作,用np.char.add替代+可规避类型错误:sex_arr = np.where(df_mpf['SEX']==1,'M','F').astype(str) df_mpf['DATAKEY'] = np.char.add(sex_arr, sex_arr)方案3:先生成性别字符串列再拼接
先把性别字符串存入DataFrame临时列,再直接拼接列,逻辑更直观:df_mpf['SEX_STR'] = df_mpf['SEX'].replace({1:'M', 0:'F'}).astype(str) df_mpf['DATAKEY'] = df_mpf['SEX_STR'] + df_mpf['SEX_STR']方案4:一次性构造拼接结果
直接在np.where里返回拼接后的字符串,避免多次生成数组:df_mpf['DATAKEY'] = np.where(df_mpf['SEX']==1, 'MM', 'FF')
内容的提问来源于stack exchange,提问作者Nick_FN
相关产品推荐
相关产品推荐

