1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
| """
| Includes test for last_valid_index.
| """
| import numpy as np
| import pytest
|
| from pandas import (
| DataFrame,
| Index,
| Series,
| date_range,
| )
|
|
| class TestFirstValidIndex:
| def test_first_valid_index_single_nan(self, frame_or_series):
| # GH#9752 Series/DataFrame should both return None, not raise
| obj = frame_or_series([np.nan])
|
| assert obj.first_valid_index() is None
| assert obj.iloc[:0].first_valid_index() is None
|
| @pytest.mark.parametrize(
| "empty", [DataFrame(), Series(dtype=object), Series([], index=[], dtype=object)]
| )
| def test_first_valid_index_empty(self, empty):
| # GH#12800
| assert empty.last_valid_index() is None
| assert empty.first_valid_index() is None
|
| @pytest.mark.parametrize(
| "data,idx,expected_first,expected_last",
| [
| ({"A": [1, 2, 3]}, [1, 1, 2], 1, 2),
| ({"A": [1, 2, 3]}, [1, 2, 2], 1, 2),
| ({"A": [1, 2, 3, 4]}, ["d", "d", "d", "d"], "d", "d"),
| ({"A": [1, np.nan, 3]}, [1, 1, 2], 1, 2),
| ({"A": [np.nan, np.nan, 3]}, [1, 1, 2], 2, 2),
| ({"A": [1, np.nan, 3]}, [1, 2, 2], 1, 2),
| ],
| )
| def test_first_last_valid_frame(self, data, idx, expected_first, expected_last):
| # GH#21441
| df = DataFrame(data, index=idx)
| assert expected_first == df.first_valid_index()
| assert expected_last == df.last_valid_index()
|
| @pytest.mark.parametrize(
| "index",
| [Index([str(i) for i in range(20)]), date_range("2020-01-01", periods=20)],
| )
| def test_first_last_valid(self, index):
| mat = np.random.default_rng(2).standard_normal(len(index))
| mat[:5] = np.nan
| mat[-5:] = np.nan
|
| frame = DataFrame({"foo": mat}, index=index)
| assert frame.first_valid_index() == frame.index[5]
| assert frame.last_valid_index() == frame.index[-6]
|
| ser = frame["foo"]
| assert ser.first_valid_index() == frame.index[5]
| assert ser.last_valid_index() == frame.index[-6]
|
| @pytest.mark.parametrize(
| "index",
| [Index([str(i) for i in range(10)]), date_range("2020-01-01", periods=10)],
| )
| def test_first_last_valid_all_nan(self, index):
| # GH#17400: no valid entries
| frame = DataFrame(np.nan, columns=["foo"], index=index)
|
| assert frame.last_valid_index() is None
| assert frame.first_valid_index() is None
|
| ser = frame["foo"]
| assert ser.first_valid_index() is None
| assert ser.last_valid_index() is None
|
|
