Python Pandas從部分字符串匹配中填充列

Question

我有一個如下所示的數據Block ，我需要根據“ Program Number ”列中表示“ _block_1”或“ _block_2”的部分字符串匹配，在其中創建一個新列“ Block ，其值為1或2 。 我一直在嘗試如果語句和.str.contains但不能使其正常工作。 你會怎么做？

148 0209-3SP_block_1    ['g76p010060q00250r.0005'   'JEBD0507160 REV A' CHNCIII
149 0209-3SP_block_2    ['g76x.3761z-.500p03067q03067f.05'  'JEBD0507160 REV A' CHNC III
150 0209-5SP_block_1    ['g76p020060q00250r.0005'   'JEBD0507160 REV A' CHNC III
151 0209-5SP_block_2    ['g76x.3767z-.48p03067q03067f.05'   'JEBD0507160 REV A' CHNC III
152 0210-3SP_block_1    ['g76p010060q00250r.0005'   'JEBD0507160 REV A' CHNC III

Answer 1

您可以使用numpy where方法：

import numpy as np

df['Block'] = np.where(
    df['Machine'].str.contains('_block_1'),1,
    np.where(df['Machine'].str.contains('_block_2'),2,0)
)

否則，假設所有字符串具有相同的長度：

df['Block'] = df['Machine'].str[15:].astype(int)