Skip to content

Commit c8e56c7

Browse files
jbrockmendelalanbato
jbrockmendel
authored and
alanbato
committed
flake8 cleanup (pandas-dev#17873)
1 parent 510bc23 commit c8e56c7

File tree

1 file changed

+77
-27
lines changed

1 file changed

+77
-27
lines changed

‎pandas/_libs/tslibs/fields.pyx

+77-27
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,9 @@ def get_start_end_field(ndarray[int64_t] dtindex, object field,
160160
if field == 'is_month_start':
161161
if is_business:
162162
for i in range(count):
163-
if dtindex[i] == NPY_NAT: out[i] = 0; continue
163+
if dtindex[i] == NPY_NAT:
164+
out[i] = 0
165+
continue
164166

165167
pandas_datetime_to_datetimestruct(
166168
dtindex[i], PANDAS_FR_ns, &dts)
@@ -172,7 +174,9 @@ def get_start_end_field(ndarray[int64_t] dtindex, object field,
172174
return out.view(bool)
173175
else:
174176
for i in range(count):
175-
if dtindex[i] == NPY_NAT: out[i] = 0; continue
177+
if dtindex[i] == NPY_NAT:
178+
out[i] = 0
179+
continue
176180

177181
pandas_datetime_to_datetimestruct(
178182
dtindex[i], PANDAS_FR_ns, &dts)
@@ -185,7 +189,9 @@ def get_start_end_field(ndarray[int64_t] dtindex, object field,
185189
elif field == 'is_month_end':
186190
if is_business:
187191
for i in range(count):
188-
if dtindex[i] == NPY_NAT: out[i] = 0; continue
192+
if dtindex[i] == NPY_NAT:
193+
out[i] = 0
194+
continue
189195

190196
pandas_datetime_to_datetimestruct(
191197
dtindex[i], PANDAS_FR_ns, &dts)
@@ -202,7 +208,9 @@ def get_start_end_field(ndarray[int64_t] dtindex, object field,
202208
return out.view(bool)
203209
else:
204210
for i in range(count):
205-
if dtindex[i] == NPY_NAT: out[i] = 0; continue
211+
if dtindex[i] == NPY_NAT:
212+
out[i] = 0
213+
continue
206214

207215
pandas_datetime_to_datetimestruct(
208216
dtindex[i], PANDAS_FR_ns, &dts)
@@ -219,7 +227,9 @@ def get_start_end_field(ndarray[int64_t] dtindex, object field,
219227
elif field == 'is_quarter_start':
220228
if is_business:
221229
for i in range(count):
222-
if dtindex[i] == NPY_NAT: out[i] = 0; continue
230+
if dtindex[i] == NPY_NAT:
231+
out[i] = 0
232+
continue
223233

224234
pandas_datetime_to_datetimestruct(
225235
dtindex[i], PANDAS_FR_ns, &dts)
@@ -232,7 +242,9 @@ def get_start_end_field(ndarray[int64_t] dtindex, object field,
232242
return out.view(bool)
233243
else:
234244
for i in range(count):
235-
if dtindex[i] == NPY_NAT: out[i] = 0; continue
245+
if dtindex[i] == NPY_NAT:
246+
out[i] = 0
247+
continue
236248

237249
pandas_datetime_to_datetimestruct(
238250
dtindex[i], PANDAS_FR_ns, &dts)
@@ -245,7 +257,9 @@ def get_start_end_field(ndarray[int64_t] dtindex, object field,
245257
elif field == 'is_quarter_end':
246258
if is_business:
247259
for i in range(count):
248-
if dtindex[i] == NPY_NAT: out[i] = 0; continue
260+
if dtindex[i] == NPY_NAT:
261+
out[i] = 0
262+
continue
249263

250264
pandas_datetime_to_datetimestruct(
251265
dtindex[i], PANDAS_FR_ns, &dts)
@@ -263,7 +277,9 @@ def get_start_end_field(ndarray[int64_t] dtindex, object field,
263277
return out.view(bool)
264278
else:
265279
for i in range(count):
266-
if dtindex[i] == NPY_NAT: out[i] = 0; continue
280+
if dtindex[i] == NPY_NAT:
281+
out[i] = 0
282+
continue
267283

268284
pandas_datetime_to_datetimestruct(
269285
dtindex[i], PANDAS_FR_ns, &dts)
@@ -280,7 +296,9 @@ def get_start_end_field(ndarray[int64_t] dtindex, object field,
280296
elif field == 'is_year_start':
281297
if is_business:
282298
for i in range(count):
283-
if dtindex[i] == NPY_NAT: out[i] = 0; continue
299+
if dtindex[i] == NPY_NAT:
300+
out[i] = 0
301+
continue
284302

285303
pandas_datetime_to_datetimestruct(
286304
dtindex[i], PANDAS_FR_ns, &dts)
@@ -293,7 +311,9 @@ def get_start_end_field(ndarray[int64_t] dtindex, object field,
293311
return out.view(bool)
294312
else:
295313
for i in range(count):
296-
if dtindex[i] == NPY_NAT: out[i] = 0; continue
314+
if dtindex[i] == NPY_NAT:
315+
out[i] = 0
316+
continue
297317

298318
pandas_datetime_to_datetimestruct(
299319
dtindex[i], PANDAS_FR_ns, &dts)
@@ -306,7 +326,9 @@ def get_start_end_field(ndarray[int64_t] dtindex, object field,
306326
elif field == 'is_year_end':
307327
if is_business:
308328
for i in range(count):
309-
if dtindex[i] == NPY_NAT: out[i] = 0; continue
329+
if dtindex[i] == NPY_NAT:
330+
out[i] = 0
331+
continue
310332

311333
pandas_datetime_to_datetimestruct(
312334
dtindex[i], PANDAS_FR_ns, &dts)
@@ -324,7 +346,9 @@ def get_start_end_field(ndarray[int64_t] dtindex, object field,
324346
return out.view(bool)
325347
else:
326348
for i in range(count):
327-
if dtindex[i] == NPY_NAT: out[i] = 0; continue
349+
if dtindex[i] == NPY_NAT:
350+
out[i] = 0
351+
continue
328352

329353
pandas_datetime_to_datetimestruct(
330354
dtindex[i], PANDAS_FR_ns, &dts)
@@ -367,7 +391,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
367391
if field == 'Y':
368392
with nogil:
369393
for i in range(count):
370-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
394+
if dtindex[i] == NPY_NAT:
395+
out[i] = -1
396+
continue
371397

372398
pandas_datetime_to_datetimestruct(
373399
dtindex[i], PANDAS_FR_ns, &dts)
@@ -377,7 +403,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
377403
elif field == 'M':
378404
with nogil:
379405
for i in range(count):
380-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
406+
if dtindex[i] == NPY_NAT:
407+
out[i] = -1
408+
continue
381409

382410
pandas_datetime_to_datetimestruct(
383411
dtindex[i], PANDAS_FR_ns, &dts)
@@ -387,7 +415,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
387415
elif field == 'D':
388416
with nogil:
389417
for i in range(count):
390-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
418+
if dtindex[i] == NPY_NAT:
419+
out[i] = -1
420+
continue
391421

392422
pandas_datetime_to_datetimestruct(
393423
dtindex[i], PANDAS_FR_ns, &dts)
@@ -397,7 +427,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
397427
elif field == 'h':
398428
with nogil:
399429
for i in range(count):
400-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
430+
if dtindex[i] == NPY_NAT:
431+
out[i] = -1
432+
continue
401433

402434
pandas_datetime_to_datetimestruct(
403435
dtindex[i], PANDAS_FR_ns, &dts)
@@ -407,7 +439,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
407439
elif field == 'm':
408440
with nogil:
409441
for i in range(count):
410-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
442+
if dtindex[i] == NPY_NAT:
443+
out[i] = -1
444+
continue
411445

412446
pandas_datetime_to_datetimestruct(
413447
dtindex[i], PANDAS_FR_ns, &dts)
@@ -417,7 +451,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
417451
elif field == 's':
418452
with nogil:
419453
for i in range(count):
420-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
454+
if dtindex[i] == NPY_NAT:
455+
out[i] = -1
456+
continue
421457

422458
pandas_datetime_to_datetimestruct(
423459
dtindex[i], PANDAS_FR_ns, &dts)
@@ -427,7 +463,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
427463
elif field == 'us':
428464
with nogil:
429465
for i in range(count):
430-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
466+
if dtindex[i] == NPY_NAT:
467+
out[i] = -1
468+
continue
431469

432470
pandas_datetime_to_datetimestruct(
433471
dtindex[i], PANDAS_FR_ns, &dts)
@@ -437,7 +475,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
437475
elif field == 'ns':
438476
with nogil:
439477
for i in range(count):
440-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
478+
if dtindex[i] == NPY_NAT:
479+
out[i] = -1
480+
continue
441481

442482
pandas_datetime_to_datetimestruct(
443483
dtindex[i], PANDAS_FR_ns, &dts)
@@ -446,7 +486,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
446486
elif field == 'doy':
447487
with nogil:
448488
for i in range(count):
449-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
489+
if dtindex[i] == NPY_NAT:
490+
out[i] = -1
491+
continue
450492

451493
pandas_datetime_to_datetimestruct(
452494
dtindex[i], PANDAS_FR_ns, &dts)
@@ -457,7 +499,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
457499
elif field == 'dow':
458500
with nogil:
459501
for i in range(count):
460-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
502+
if dtindex[i] == NPY_NAT:
503+
out[i] = -1
504+
continue
461505

462506
pandas_datetime_to_datetimestruct(
463507
dtindex[i], PANDAS_FR_ns, &dts)
@@ -467,7 +511,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
467511
elif field == 'woy':
468512
with nogil:
469513
for i in range(count):
470-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
514+
if dtindex[i] == NPY_NAT:
515+
out[i] = -1
516+
continue
471517

472518
pandas_datetime_to_datetimestruct(
473519
dtindex[i], PANDAS_FR_ns, &dts)
@@ -477,7 +523,7 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
477523
doy = mo_off + dts.day
478524
dow = dayofweek(dts.year, dts.month, dts.day)
479525

480-
#estimate
526+
# estimate
481527
woy = (doy - 1) - dow + 3
482528
if woy >= 0:
483529
woy = woy / 7 + 1
@@ -498,7 +544,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
498544
elif field == 'q':
499545
with nogil:
500546
for i in range(count):
501-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
547+
if dtindex[i] == NPY_NAT:
548+
out[i] = -1
549+
continue
502550

503551
pandas_datetime_to_datetimestruct(
504552
dtindex[i], PANDAS_FR_ns, &dts)
@@ -509,7 +557,9 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
509557
elif field == 'dim':
510558
with nogil:
511559
for i in range(count):
512-
if dtindex[i] == NPY_NAT: out[i] = -1; continue
560+
if dtindex[i] == NPY_NAT:
561+
out[i] = -1
562+
continue
513563

514564
pandas_datetime_to_datetimestruct(
515565
dtindex[i], PANDAS_FR_ns, &dts)
@@ -522,7 +572,7 @@ def get_date_field(ndarray[int64_t] dtindex, object field):
522572

523573

524574
cdef inline int days_in_month(pandas_datetimestruct dts) nogil:
525-
return days_per_month_table[is_leapyear(dts.year)][dts.month -1]
575+
return days_per_month_table[is_leapyear(dts.year)][dts.month - 1]
526576

527577

528578
cpdef isleapyear_arr(ndarray years):

0 commit comments

Comments
 (0)