python 提取datafarm中某列符合条件的值

Python提供了很多方法来提取DataFrame中某列符合条件的值。以下是其中一些方法：

1. 使用loc方法：

import pandas as pd

df = pd.DataFrame({'col1': [1, 2, 3, 4], 'col2': ['a', 'b', 'a', 'c']})
result = df.loc[df['col2'] == 'a', 'col1']
print(result)

这将提取df中“col2”列符合条件“a”的所有值，并将结果存储在变量“result”中。

2. 使用iloc方法：

import pandas as pd

df = pd.DataFrame({'col1': [1, 2, 3, 4], 'col2': ['a', 'b', 'a', 'c']})
result = df.iloc[df['col2'] == 'a', 0]
print(result)

这将提取df中“col2”列符合条件“a”的所有值，并将结果存储在变量“result”中。

3. 使用query方法：

import pandas as pd

df = pd.DataFrame({'col1': [1, 2, 3, 4], 'col2': ['a', 'b', 'a', 'c']})
result = df.query('col2 == "a"')['col1']
print(result)

这将提取df中“col2”列符合条件“a”的所有值，并将结果存储在变量“result”中。

无论您使用哪种方法，都可以轻松提取DataFrame中某列符合条件的值。