使用熊貓：import pandas as pd

id1 = pd.read_csv('id1.txt')

id2 = pd.read_csv('id2.txt')

df = id1.merge(id2.sort_values(by='ID2').drop_duplicates('ID2').rename(columns={'ID2':'ID1'}))

print(df)

產生：

^{pr2}$

對于大型數據集，您可能需要執行以下操作：# [Optional] sort locations and drop duplicates

id2.sort_values(by='ID2', inplace=True)

id2.drop_duplicates('ID2', inplace=True)

# columns that you are merging must have the same name

id2.rename(columns={'ID2':'ID1'}, inplace=True)

# perform the merge

df = id1.merge(id2)

如果沒有drop-duplicates，則每個項目都有一行：df = id1.merge(id2.rename(columns={'ID2':'ID1'}))

print(id2)

print(df)

給予：ID2 RA DEC

0 101 4.5 10.5

1 107 90.1 55.5

2 102 30.5 3.3

3 103 60.1 40.6

4 104 10.8 5.6

5 103 60.1 40.6

6 104 10.9 5.6

ID1 z e PA n RA DEC

0 101 1.0 1.2 1.5 1.8 4.5 10.5

1 104 1.5 1.8 2.2 3.1 10.8 5.6

2 104 1.5 1.8 2.2 3.1 10.9 5.6

請注意，此解決方案保留了列的不同類型：>>> id1.ID1.dtype

dtype('int64')

>>> id1[' z'].dtype

dtype('float64')

由于在標題行中逗號后面有空格，因此這些空格成為列名的一部分，因此需要使用id1['z']引用第二列。通過修改read語句，不再需要執行以下操作：>>> id1 = pd.read_csv('id1.txt', skipinitialspace=True)

>>> id1.z.dtype

dtype('float64')

總結

以上是生活随笔為你收集整理的python查找两个数组中相同的元素_匹配两个numpy数组以找到相同的元素的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。