4桁は取り出さないけど2桁か1桁の数値だったら取り出す正規表現を教えてください

Question

Pythonで文字列から数値が4桁だったら取り出さないけど、2桁か1桁だったら取り出すコードを書きたいです。
Pythonの正規表現では\dで数値を表せます。
4桁を取り出す場合はこんな感じです。

import re

text = '2001年宇宙の旅'
p = re.compile(r'\d\d\d\d')
result = p.search(text)
if result is not None:
    print(result.group())

>>> 2001

text = '2001年7月宇宙の旅'
# ここから7だけを取り出したい

Answer 1

先読み言明・あと読み言明を使えばすんなり書けると思います。言明とは文字ではなく「位置」にマッチする正規表現と考えれば解りやすいです。
参考：正規表現の先読み・後読みを極める！

「左側に数字が来なくて、数字が1個か2個だけ並び、その右に数字が来ない」なら、

(?<!\d)\d{1,2}(?!\d)

ということで、

# coding:utf-8
import re

text = '2001年7月宇宙の旅'
p = re.compile(r'(?<!\d)\d{1,2}(?!\d)')
result = p.search(text)
if result is not None:
    print(result.group())

Comments

[otn]

否定先読み・後読みにすれば、先頭・末尾でもいいんですね！なるほど。

[Shichi7]

一番丁寧に教えてくださったのでベストアンンサーにさせていただきます。
みなさんありがとうございました。

Answer 2

(^|\D)(\d{1,2})($|\D)
group(2)で抽出できます。

\d{m}このmを書き換えれば、任意長の文字が取れますね。

Answer 3

import re

text = '2001年7月宇宙の旅'
p = re.compile(r'(?<=\D)\d{1,2}(?=\D)')
result = p.search(' '+text+' ')
if result is not None:
    print(result.group())

(?<=^|\D) や (?=$|\D)というのがエラーになってしまうので、前後に非数字を付けました。

Answer 4

4桁の数字を事前に*等で置き換えてから、1or2桁を抽出するっていうのをよくやります。

追記
3桁ってどうするの？

Answer 5

先に数値を全部取り出してから1桁or2桁を絞り込む，という方法はありですか？

import re
text = "2001年7月宇宙の旅"
pattern = re.compile(r'\d+')
list(filter(lambda x: len(x) in [1, 2], re.findall(pattern, text)))
# ['7']

# または

list(filter(lambda x: int(x) > 0 and int(x) < 100, re.findall(p, text)))
# ['7']

Answer 6

^\d{1,2}$

でマッチするかと

ref:

• python - Regex to match digits of specific length - Stack Overflow

追記

コメントでダメ出しされたのでリベンジ。

(^(\d{1,2})$|[^\d](\d{1,2}))

これで$1/$3をみる・・・いやダサいな。

Comments

[KojiDoi]

これだとテキストの途中に数値が入っているパターンに対応できないのでは…

[yumetodo]

あ・・・だめだこれ

Answer 7

print(re.search(r'[0-9][0-9]?',text))

こちらでどうでしょう？

追記
これでは4桁中の数字も含まれてしまうのでもしかすると質問の意に沿っていないかもしれません