[已解决]pandas按照条件判断填充一列的内容

颂风侯

放了一个附件，请求大佬们帮助。
内容是这样的。文本表格格式如下：
管线号                                                    焊缝编号        焊接类型        壁厚        材质
天然气80-NG-5310-10001-A0BE31-N(GC2)        1                  对焊        3.05        S31603&S31603
天然气80-NG-5310-10001-A0BE31-N(GC2)        2                  对焊        3.05        20
天然气80-NG-5310-10001-A0BE31-N(GC2)        3                 对焊                3.05        S31603
天然气80-NG-5310-10001-A0BE31-N(GC2)        4                 对焊               3.05        S31603&S31603
天然气80-NG-5310-10001-A0BE31-N(GC2)        5                 对焊               3.05        S31603&S31603
天然气80-NG-5310-10001-A0BE31-N(GC2)        6                 对焊              3.05        S31603&S31603
天然气80-NG-5310-10001-A0BE31-N(GC2)        7                 对焊              3.05        S31603&S31603
污水池甲烷气管线50-PG1-5310-10001-A0BE31-N(GC2)        1        对焊        3.05        S31603&S31603

我需要做到的是这样：
根据列"焊接类型","材质"，“壁厚”的值依次判断，我用的pycharm，第三方用的pandas。
判断原则如下
1，如果是对焊，并且材质的文本内容包含“12Cr”，如果“壁厚”大于13，列“工艺卡编号”的值为"HKB12-06"
2,如果是对焊，并且材质的文本内容包含“12Cr”，如果“壁厚”大于10，列“工艺卡编号”的值为"HKB12-05"
3,如果是对焊，并且材质的文本内容包含“12Cr”，如果“壁厚”大于8，列“工艺卡编号”的值为"HKB12-04"
4,如果是承插焊，并且材质的文本内容包含“12Cr”，如果“壁厚”大于5.5，列“工艺卡编号”的值为"HKC12-03"
5,如果是承插焊，并且材质的文本内容包含“12Cr”，如果“壁厚”大于3，列“工艺卡编号”的值为"HKC12-02"

1，如果是对焊，并且材质的文本内容包含“20”，如果“壁厚”大于13，列“工艺卡编号”的值为"HKB20-06"
2,如果是对焊，并且材质的文本内容包含“20”，如果“壁厚”大于10，列“工艺卡编号”的值为"HKB20-05"
3,如果是对焊，并且材质的文本内容包含“20”，如果“壁厚”大于8，列“工艺卡编号”的值为"HKB20-04"
4,如果是承插焊，并且材质的文本内容包含“20，如果“壁厚”大于5.5，列“工艺卡编号”的值为"HKC20-03"
5,如果是承插焊，并且材质的文本内容包含“20”，如果“壁厚”大于3，列“工艺卡编号”的值为"HKC20-02"
以上条件都是一个成立了就不再执行其他判断
求助怎么实现呢

最佳答案

月排行榜 / 总排行榜

FishC

2024-9-12 12:56:49

颂风侯 发表于 2024-09-12 12:56
不想用numpy函数，还是想用熟悉层次的if函数可以吗？
...

当然可以使用 `apply` 和自定义函数来实现这个需求。以下是一个示例代码，展示如何根据不同条件填充“工艺卡编号”列：

1. import pandas as pd
   
2. 
   
3. # 创建示例数据框
   
4. data = {
   
5.     '管线号': [
   
6.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
   
7.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
   
8.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
   
9.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
   
10.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
    
11.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
    
12.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
    
13.         '污水池甲烷气管线50-PG1-5310-10001-A0BE31-N(GC2)'
    
14.     ],
    
15.     '焊缝编号': [1, 2, 3, 4, 5, 6, 7, 1],
    
16.     '焊接类型': ['对焊', '对焊', '对焊', '对焊', '对焊', '对焊', '对焊', '对焊'],
    
17.     '壁厚': [3.05, 3.05, 3.05, 3.05, 3.05, 3.05, 3.05, 3.05],
    
18.     '材质': ['S31603&S31603', '20', 'S31603', 'S31603&S31603', 'S31603&S31603', 'S31603&S31603', 'S31603&S31603', 'S31603&S31603']
    
19. }
    
20. 
    
21. df = pd.DataFrame(data)
    
22. 
    
23. # 定义填充“工艺卡编号”的函数
    
24. def fill_process_card(row):
    
25.     if row['焊接类型'] == '对焊' and '12Cr' in row['材质']:
    
26.         if row['壁厚'] > 13:
    
27.             return 'HKB12-06'
    
28.         elif row['壁厚'] > 10:
    
29.             return 'HKB12-05'
    
30.         elif row['壁厚'] > 8:
    
31.             return 'HKB12-04'
    
32.     elif row['焊接类型'] == '承插焊' and '12Cr' in row['材质']:
    
33.         if row['壁厚'] > 5.5:
    
34.             return 'HKC12-03'
    
35.         elif row['壁厚'] > 3:
    
36.             return 'HKC12-02'
    
37.     elif row['焊接类型'] == '对焊' and '20' in row['材质']:
    
38.         if row['壁厚'] > 13:
    
39.             return 'HKB20-06'
    
40.         elif row['壁厚'] > 10:
    
41.             return 'HKB20-05'
    
42.         elif row['壁厚'] > 8:
    
43.             return 'HKB20-04'
    
44.     elif row['焊接类型'] == '承插焊' and '20' in row['材质']:
    
45.         if row['壁厚'] > 5.5:
    
46.             return 'HKC20-03'
    
47.         elif row['壁厚'] > 3:
    
48.             return 'HKC20-02'
    
49.     return None
    
50. 
    
51. # 应用函数填充“工艺卡编号”列
    
52. df['工艺卡编号'] = df.apply(fill_process_card, axis=1)
    
53. 
    
54. print(df)

复制代码

这个代码定义了一个 `fill_process_card` 函数，根据你提供的条件来填充“工艺卡编号”列。然后使用 `apply` 方法将这个函数应用到数据框的每一行。这样可以确保代码逻辑清晰且易于理解。

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。

跳转到最佳答案楼层

FishC

鱼油，你好！根据你的需求，可以使用Pandas库来实现。以下是一个示例代码，展示如何根据不同条件填充“工艺卡编号”列：

1. import pandas as pd
   
2. 
   
3. # 创建示例数据框
   
4. data = {
   
5.     '管线号': ['天然气80-NG-5310-10001-A0BE31-N(GC2)']*7 + ['污水池甲烷气管线50-PG1-5310-10001-A0BE31-N(GC2)'],
   
6.     '焊缝编号': [1, 2, 3, 4, 5, 6, 7, 1],
   
7.     '焊接类型': ['对焊']*8,
   
8.     '壁厚': [3.05]*8,
   
9.     '材质': ['S31603&S31603', '20', 'S31603', 'S31603&S31603', 'S31603&S31603', 'S31603&S31603', 'S31603&S31603', 'S31603&S31603']
   
10. }
    
11. 
    
12. df = pd.DataFrame(data)
    
13. 
    
14. # 定义条件和对应的值
    
15. conditions = [
    
16.     (df['焊接类型'] == '对焊') & (df['材质'].str.contains('12Cr')) & (df['壁厚'] > 13),
    
17.     (df['焊接类型'] == '对焊') & (df['材质'].str.contains('12Cr')) & (df['壁厚'] > 10),
    
18.     (df['焊接类型'] == '对焊') & (df['材质'].str.contains('12Cr')) & (df['壁厚'] > 8),
    
19.     (df['焊接类型'] == '承插焊') & (df['材质'].str.contains('12Cr')) & (df['壁厚'] > 5.5),
    
20.     (df['焊接类型'] == '承插焊') & (df['材质'].str.contains('12Cr')) & (df['壁厚'] > 3),
    
21.     (df['焊接类型'] == '对焊') & (df['材质'].str.contains('20')) & (df['壁厚'] > 13),
    
22.     (df['焊接类型'] == '对焊') & (df['材质'].str.contains('20')) & (df['壁厚'] > 10),
    
23.     (df['焊接类型'] == '对焊') & (df['材质'].str.contains('20')) & (df['壁厚'] > 8),
    
24.     (df['焊接类型'] == '承插焊') & (df['材质'].str.contains('20')) & (df['壁厚'] > 5.5),
    
25.     (df['焊接类型'] == '承插焊') & (df['材质'].str.contains('20')) & (df['壁厚'] > 3)
    
26. ]
    
27. 
    
28. choices = [
    
29.     'HKB12-06',
    
30.     'HKB12-05',
    
31.     'HKB12-04',
    
32.     'HKC12-03',
    
33.     'HKC12-02',
    
34.     'HKB20-06',
    
35.     'HKB20-05',
    
36.     'HKB20-04',
    
37.     'HKC20-03',
    
38.     'HKC20-02'
    
39. ]
    
40. 
    
41. # 使用 numpy 的 select 函数来根据条件选择相应的值
    
42. df['工艺卡编号'] = pd.Series(pd.NA)
    
43. import numpy as np
    
44. df['工艺卡编号'] = np.select(conditions, choices, default=None)
    
45. 
    
46. print(df)

复制代码

此代码利用了`numpy.select`函数来根据多个条件应用相应的选择。请根据你的实际数据更新data部分，并确保已安装了`pandas`和`numpy`库：

1. pip install pandas numpy

复制代码

希望以上代码能帮到你！有什么问题随时问哦！

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。

颂风侯

不想用numpy函数，还是想用熟悉层次的if函数可以吗？
比如做个函数，进行填充
比如下面这样看起来清楚点
def city_type(x)：
   if x['城市]in(['北京',上海',深圳']):
       return'一线'
   else:
       return'新一线
df['city type']=df.apply(city_type,axis=1)

FishC

颂风侯 发表于 2024-09-12 12:56
不想用numpy函数，还是想用熟悉层次的if函数可以吗？
...

当然可以使用 `apply` 和自定义函数来实现这个需求。以下是一个示例代码，展示如何根据不同条件填充“工艺卡编号”列：

1. import pandas as pd
   
2. 
   
3. # 创建示例数据框
   
4. data = {
   
5.     '管线号': [
   
6.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
   
7.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
   
8.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
   
9.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
   
10.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
    
11.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
    
12.         '天然气80-NG-5310-10001-A0BE31-N(GC2)',
    
13.         '污水池甲烷气管线50-PG1-5310-10001-A0BE31-N(GC2)'
    
14.     ],
    
15.     '焊缝编号': [1, 2, 3, 4, 5, 6, 7, 1],
    
16.     '焊接类型': ['对焊', '对焊', '对焊', '对焊', '对焊', '对焊', '对焊', '对焊'],
    
17.     '壁厚': [3.05, 3.05, 3.05, 3.05, 3.05, 3.05, 3.05, 3.05],
    
18.     '材质': ['S31603&S31603', '20', 'S31603', 'S31603&S31603', 'S31603&S31603', 'S31603&S31603', 'S31603&S31603', 'S31603&S31603']
    
19. }
    
20. 
    
21. df = pd.DataFrame(data)
    
22. 
    
23. # 定义填充“工艺卡编号”的函数
    
24. def fill_process_card(row):
    
25.     if row['焊接类型'] == '对焊' and '12Cr' in row['材质']:
    
26.         if row['壁厚'] > 13:
    
27.             return 'HKB12-06'
    
28.         elif row['壁厚'] > 10:
    
29.             return 'HKB12-05'
    
30.         elif row['壁厚'] > 8:
    
31.             return 'HKB12-04'
    
32.     elif row['焊接类型'] == '承插焊' and '12Cr' in row['材质']:
    
33.         if row['壁厚'] > 5.5:
    
34.             return 'HKC12-03'
    
35.         elif row['壁厚'] > 3:
    
36.             return 'HKC12-02'
    
37.     elif row['焊接类型'] == '对焊' and '20' in row['材质']:
    
38.         if row['壁厚'] > 13:
    
39.             return 'HKB20-06'
    
40.         elif row['壁厚'] > 10:
    
41.             return 'HKB20-05'
    
42.         elif row['壁厚'] > 8:
    
43.             return 'HKB20-04'
    
44.     elif row['焊接类型'] == '承插焊' and '20' in row['材质']:
    
45.         if row['壁厚'] > 5.5:
    
46.             return 'HKC20-03'
    
47.         elif row['壁厚'] > 3:
    
48.             return 'HKC20-02'
    
49.     return None
    
50. 
    
51. # 应用函数填充“工艺卡编号”列
    
52. df['工艺卡编号'] = df.apply(fill_process_card, axis=1)
    
53. 
    
54. print(df)

复制代码

这个代码定义了一个 `fill_process_card` 函数，根据你提供的条件来填充“工艺卡编号”列。然后使用 `apply` 方法将这个函数应用到数据框的每一行。这样可以确保代码逻辑清晰且易于理解。

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。