Pandas截取列部分字符,并据此修改另一列的数据
#截取'股票代码'第一个字符 df['首字符'] = df['股票代码'].str[0]
#添加市场代码列,初始值为0=深圳
df['市场'] = 0
#根据一列的值,修改另一列的值
df.loc[df['首字符'] == '6','市场'] = '1' # 根据'首字符'列的值,修改'市场'的值。 1表示上海
或者用下面的代码:
1 df.loc[df.股票代码.str[0] == '6','市场'] = '1' # 修改'市场'的值。 1表示上海 2 3 dX.loc[df.股票代码.str[0] != '6','市场'] = '0' # 修改'市场'的值。 0表示深圳