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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
import numpy as np
from Orange.data import Table
from Orange.widgets import widget, gui
from Orange.widgets.utils import itemmodels
from Orange.widgets.utils.localization import pl
from Orange.widgets.utils.sql import check_sql_input
from Orange.widgets.utils.widgetpreview import WidgetPreview
from Orange.widgets.widget import Input, Output
from Orange.widgets.utils.annotated_data import (create_annotated_table)
class OWSelectByDataIndex(widget.OWWidget):
name = "Select by Data Index"
description = "Match instances by index from data subset."
category = "Transform"
icon = "icons/SelectByDataIndex.svg"
priority = 1112
class Inputs:
data = Input("Data", Table)
data_subset = Input("Data Subset", Table)
class Outputs:
matching_data = Output("Matching Data", Table, replaces=["Data"], default=True)
non_matching_data = Output("Unmatched Data", Table)
# avoiding the default annotated output name (Data), as it was used
# for Matching Data previously
annotated_data = Output("Annotated Data", Table)
want_main_area = False
buttons_area_orientation = None
resizing_enabled = False
class Warning(widget.OWWidget.Warning):
instances_not_matching = widget.Msg("Input tables do not share any instances.")
def __init__(self):
super().__init__()
self.data = None
self.data_subset = None
self.model = itemmodels.VariableListModel()
self.model_unique_with_id = itemmodels.VariableListModel()
self.extra_model_unique = itemmodels.VariableListModel()
self.extra_model_unique_with_id = itemmodels.VariableListModel()
box = gui.widgetBox(self.controlArea, True)
gui.label(
box, self, """
Data rows keep their identity even when some or all original variables
are replaced by variables computed from the original ones.
This widget gets two data tables ("Data" and "Data Subset") that
can be traced back to the same source. It selects all rows from Data
that appear in Data Subset, based on row identity and not actual data.
""".strip(), box=True)
@Inputs.data
@check_sql_input
def set_data(self, data):
self.data = data
@Inputs.data_subset
@check_sql_input
def set_data_subset(self, data):
self.data_subset = data
def handleNewSignals(self):
self._invalidate()
def commit(self):
self.Warning.instances_not_matching.clear()
if not self.data:
matching_output = None
non_matching_output = None
annotated_output = None
else:
subset_ids = []
if self.data_subset is not None:
subset_ids = self.data_subset.ids
if not np.intersect1d(subset_ids, self.data.ids).size:
self.Warning.instances_not_matching()
row_sel = np.isin(self.data.ids, subset_ids)
matching_output = self.data[row_sel]
non_matching_output = self.data[~row_sel]
annotated_output = create_annotated_table(self.data, row_sel)
self.Outputs.matching_data.send(matching_output)
self.Outputs.non_matching_data.send(non_matching_output)
self.Outputs.annotated_data.send(annotated_output)
def _invalidate(self):
self.commit()
def send_report(self):
def data_info_text(data):
if data is None:
return "No data."
nvars = len(data.domain.variables) + len(data.domain.metas)
return f"{data.name}, " \
f"{len(data)} {pl(len(data), 'instance')}, " \
f"{nvars} {pl(nvars, 'variable')}"
self.report_items("",
[("Data", data_info_text(self.data)),
("Data Subset", data_info_text(self.data_subset))])
if __name__ == "__main__": # pragma: no cover
iris = Table("iris.tab")
WidgetPreview(OWSelectByDataIndex).run(
set_data=iris,
set_data_subset=iris[:20])
|
