如何在python中实现where子句

时间:2018-04-07 18:40:25

标签: sql python-3.x pandas where-clause

我想使用Python复制where子句在SQL中的作用。很多时候where子句中的条件可能很复杂并且具有多个条件。我可以通过以下方式完成。但我认为应该有一种更聪明的方法来实现这一目标。我有以下数据和代码。

我的要求是:我想只在地址中的第一个字母是' N'时才选择所有列。这是初始数据框架。

enter image description here

d = {'name': ['john', 'tom', 'bob', 'rock', 'dick'], 'Age': [23, 32, 45, 42, 28], 'YrsOfEducation': [10, 15, 8, 12, 10], 'Address': ['NY', 'NJ', 'PA', 'NY', 'CA']}
import pandas as pd
df = pd.DataFrame(data = d)
df['col1'] = df['Address'].str[0:1] #creating a new column which will have only the first letter from address column
n = df['col1'] == 'N' #creating a filtering criteria where the letter will be equal to N
newdata = df[n] # filtering the dataframe 
newdata1 = newdata.drop('col1', axis = 1) # finally dropping the extra column 'col1'

所以经过7行代码我得到了这个输出:

enter image description here

我的问题是我怎样才能更有效地做到这一点,还是有更明智的方法来做到这一点?

1 个答案:

答案 0 :(得分:4)

不需要新列:

int i = 0, x = 0;
while (x < 30) {
    x++;
    if (13 < i++) break;
    cout << i << endl;
}