我在大学期间主要使用 MATLAB 作为数学专业,我的编程只是构建数学方程和建模。现在我一直在学习使用 Python,尤其是 Pandas。我试图在一个数据帧的一列中搜索值,并将它们与不同数据帧的一列中的值进行匹配。如果它们匹配,我希望它们给原始数据框一个标签。
例如,我有我的第一列员工,我想弄清楚 aliceB 在我的第一个数据框中是忙碌还是非忙碌,并在 col3 中进行标签。
df1 = {"col1":["aliceA", "aliceB", "aliceC"], "col2":["CO", "WA", "PA"]}
df1 = pd.DataFrame(df1)
df1['col3'] = np.nan
In[]df1
Out[]:
col1 col2 col3
0 aliceA CO NaN
1 aliceB WA NaN
2 aliceC PA NaN
df2 = {'col1': ["aliceB", "aliceA", "aliceC", "bobC", "bobB", "bobA",], 'col2': ['Busy','Non-Busy','Busy','Non-Busy','Non-Busy','Busy']}
df2 = pd.DataFrame(df2)
In[]df2
Out[]:
col1 col2
0 aliceB Busy
1 aliceA Non-Busy
2 aliceC Busy
3 bobC Non-Busy
4 bobB Non-Busy
5 bobA Busy
***Preferred Output***
Out[]:
col1 col2 col3
0 aliceA CO Non-Busy
1 aliceB WA Busy
2 aliceC PA Busy
对于这种问题 MATLAB,我将使用我的两个矩阵并使用嵌套的 for 循环进行迭代以查找值。在 Python 中我做了:
for i in range(0, df2.shape[0]):
for j in range(0, df1.shape[0]):
if(df2.col1[i] == df1.col1[j]):
df1.col3[j] = df2.col2[i]
但是我收到了这个警告,我必须 Control + C 才能摆脱它才能继续:
SettingWithCopyWarning: A value is trying to be set on a copy of a slice from a DataFrame
df1
Out[]:
col1 col2 col3
0 aliceA CO Non-Busy
1 aliceB WA Busy
2 aliceC PA Busy
从技术上讲,此代码有效并且我的数据已填充,但我知道这可能不是解决我的问题的糟糕方法。对于这个小例子,它不会强迫我使用 Control+C,但是当我的 df1 有数千行长时它会这样做。
Cats萌萌
慕斯709654
相关分类