代码之家 › 专栏 › 技术社区 › lte__

Python:检查列表中是否有子字符串的优雅解决方案?

string python

lte__ · 技术社区 · 8 年前

在你结束这篇文章之前,请试着理解这个问题-这有点不同,但看起来有很多被否决的问题。

我正在寻找一种简单的方法来检查字符串的子字符串是否在字符串列表中。输入如下 2018_JAN_BGD ,我想将其中的月份信息转换为(年)季度信息。例如,Q1将是1月、2月和3月。所以我在寻找一种方法来检查 2018年1月11日 包含 JAN , FEB 或 MAR . 我可以用or写12(全年)if-else子句来实现这一点,但我觉得Python中有一个更优雅的解决方案。我正在尝试:

def tellQuarter(month):
    Q1 = ('JAN','FEB','MAR')
    Q2 = ('APR','MAY','JUN')
    Q3 = ('JUL','AUG','SEP')
    Q4 = ('OCT','NOV','DEC')
    if any(for mon in Q1 in month):
        return ("Q1")
    if any(for mon in Q2 in month):
        return ("Q2")

这当然是不正确的(无效语法),但我不知道如何正确地表达它。基本上,我想循环 Q1 ,并检查是否有任何字符串是 month .

我是说我能做到

for mon in Q1:
    if mon in month:
        return "Q1"

但是,难道没有一种更优雅的单线解决方案吗?

3 回复 | 直到 8 年前

Lanting 8 年前

作弊,但使用字典,将月份映射到季度:

monthsToQuarters = {
    'JAN': 'Q1','FEB': 'Q1','MAR': 'Q1',
    'APR': 'Q2','MAY': 'Q2','JUN': 'Q2',
    'JUL': 'Q3','AUG': 'Q3','SEP': 'Q3',
    'OCT': 'Q4','NOV': 'Q4','DEC': 'Q4'}

为您提供:

>>> monthsToQuarters['JAN']
'Q1'

您可以使用以下方法删除月份:

>>> '2018_JAN_BGD'.split('_')[1]
'JAN'

javidcf 8 年前

避免使用dict重复逻辑:

QUARTERS = {
    'Q1': ('JAN', 'FEB', 'MAR'),
    'Q2': ('APR', 'MAY', 'JUN'),
    'Q3': ('JUL', 'AUG', 'SEP'),
    'Q4': ('OCT', 'NOV', 'DEC'),
}

def tellQuarter(month):
    for q, months in QUARTERS.items():
        if any(m for m in months if m in month):
            return q
    raise ValueError("String '{}' does not contain a month.".format(month))

print(tellQuarter('2018_JAN_BGD'))
>>> Q1

Rakesh 8 年前

列表理解中出现语法错误

def tellQuarter(month):
        Q1 = ('JAN','FEB','MAR')
        Q2 = ('APR','MAY','JUN')
        Q3 = ('JUL','AUG','SEP')
        Q4 = ('OCT','NOV','DEC')
        if any(mon in month for mon in Q1):
            return ("Q1")
        if any(mon in month for mon in Q2):
            return ("Q2")

print tellQuarter("2018_JAN_BGD")

输出 :

Q1

推荐文章

Aidan · 为什么在将最后一个索引分配给空终止符(\0)后,我无法释放char缓冲区

1 年前

Ben · 统计向量中的单词在字符串中出现的频率

1 年前

Mehan Alavi · TCP Socket-打印字符串中所有不在C中的字符不起作用

1 年前

Haru Hoshizora · 为什么一个整数的位置没有改变,但值却不同

1 年前

bear_525 · 从列中删除中间名和首字母,并保存在单独的列中

1 年前

asdfadf · 为什么具有相同内存值的字符串和整数打印方式不同?

1 年前

DrakeMurdoch · 统计每行中的字符数,如果所有字符数都低于某个数字,则删除

1 年前

user764754 · 防止多行原始字符串文字中出现新行字符

1 年前

Bogaso · 从列表中返回与模式匹配的元素

1 年前

Jasco · 如何使用VBA提取两个相似字符之间的字符串中的单词?

1 年前